LSTMCell "Batch" Dimension

1

https://pytorch.org/docs/stable/nn.html#lstmcell

  1. Where can we set the “Depth” of the LSTMCell found in Stacked LSTMs?
  2. Is the batch dimension here like sentence length? How do we setup variable length ‘batch’ then?

The data is (N, nL, nE) where N is number of sentences, nL is a sentence length (Variable) and nE is the dimension of each word.

How do i setup the LSTM cell for this?