| Titre | pytorch pytorch (in torch.lstm_cell) torch 2.6.0 Segmentation Fault |
|---|
| Description | # 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)
``` |
|---|
| La source | ⚠️ https://github.com/pytorch/pytorch/issues/149626 |
|---|
| Utilisateur | Default436352 (UID 81891) |
|---|
| Soumission | 20/03/2025 16:26 (il y a 1 Année) |
|---|
| Modérer | 30/03/2025 19:42 (10 days later) |
|---|
| Statut | Accepté |
|---|
| Entrée VulDB | 302050 [PyTorch 2.6.0 torch.lstm_cell buffer overflow] |
|---|
| Points | 20 |
|---|