Batch first in LSTM
Open
- Dominant language
- Python
- Stars
- 733
- Forks
- 176
- PR merge metrics
- No merged PRs in 30d
Description
In the[ LSTM documentation](
https://pytorch.org/docs/stable/nn.html#torch.nn.LSTM) is specified that we should feed to the RNN inputs of shape (seq_len, batch, input_size) however it seems to me that we are feeding inputs with shape (batch, seq_len, input_size).
Therefore I believe that the parameter **batch_first** should be set **True**.
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.