I am trying to implement custom LSTM layer with custom cell.
It is working OK when I pass only one sample, but when I want to pass a batch of data a problem appear.
If my data look like [128, 64] where 128 is max sequence length and 64 is word embedding size, the model works fine
I want to know how to handle a batch of data e.g. [16, 128, 64] ?
I search for the implementation of the lstm cell in pytorch , but I can’t find the code that handles the batch
and this code leads to this impl.
is pytorch for loop the data batch?
Thanks in advance.