The call of LSTM may be different from official documentation of Pytorch

[chaoxianluo]

Hello, I have cloned your code and run successfully using trained model you provide. I got resonable results. However I noticed that the call of LSTM in your implentation is different from what offical documentation says.
According to official documentation, the input of LSTM should be of shape (seq_len, batch, input_size), however in your code, it's of shape (batch_size, seq_len, input_size).
I wonder whether I referenced documentation with mismatched version or there may be some error in your code.
Thanks for the share of your implementation.

ref documentation: https://pytorch.org/docs/master/generated/torch.nn.LSTM.html

[DexterFixxor]

Hello,

As you can see in the documentation, in the list of arguments there is "batch_first". That explains it.

[chaoxianluo]

Hello,

As you can see in the documentation, in the list of arguments there is "batch_first". That explains it.

OK, I notice it, thank you very much!