If i have nn.LSTM with hidden size of 400, one layer, and batch_first = True, then h_0 and c_0 inputs should be of shape (batch_size,1,400), or do i not understand something correcty? Since it throws me an error:
Expected hidden[0] size (1, 2, 400), got (2, 1, 400) (My batch_size is 2)