Building LSTM

tensorflow

#1

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

Can I add that under RNN in https://github.com/malmaud/TensorFlow.jl/blob/master/docs/src/ops.md ?

@malmaud


#2

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.