| 标题 | pytorch pytorch (in torch.lstm_cell) torch 2.6.0 Segmentation Fault |
|---|
| 描述 | # Bug
Segmentation Fault in torch.lstm_cell with Zero-Batch Inputs
# To Reproduce
```
import torch
inp = torch.full((0, 8), 0, dtype=torch.float)
hx = torch.full((0, 9), 0, dtype=torch.float)
cx = torch.full((0, 9), 0, dtype=torch.float)
w_ih = torch.full((1, 8), 1.251e+12, dtype=torch.float)
w_hh = torch.full((1, 9), 1.4013e-45, dtype=torch.float)
b_ih = None
b_hh = None
torch.lstm_cell(inp, (hx, cx), w_ih, w_hh, b_ih, b_hh)
``` |
|---|
| 来源 | ⚠️ https://github.com/pytorch/pytorch/issues/149626 |
|---|
| 用户 | Default436352 (UID 81891) |
|---|
| 提交 | 2025-03-20 16時26分 (1 年前) |
|---|
| 管理 | 2025-03-30 19時42分 (10 days later) |
|---|
| 状态 | 已接受 |
|---|
| VulDB条目 | 302050 [PyTorch 2.6.0 torch.lstm_cell 内存损坏] |
|---|
| 积分 | 20 |
|---|