Building LSTM



How do I build the LSTM cell? Is it nn.rnn.LSTMCell(input, state, input_dim=-1) ?

Can I add that under RNN in ?



Hi @malmaud, I found cell = nn.rnn_cell.LSTMCell(10) in your tests, but I don’t think anywhere else.

I’m still a little uncertain about the form of the arguments, will study the code further.

Can I somehow do cell = nn.rnn_cell.LSTMCell(options.num_units, state_is_tuple = True, forget_bias = 0.0)? The format of the arguments don’t have to be this way.