Thursday, May 25, 2017

LSTM APIs in Tensorflow

  • BasicRNNCell – A vanilla RNN cell.
  • GRUCell – A Gated Recurrent Unit cell.
  • BasicLSTMCell – An LSTM cell based on Recurrent Neural Network Regularization. No peephole connection or cell clipping.
  • LSTMCell – A more complex LSTM cell that allows for optional peephole connections and cell clipping.
  • MultiRNNCell – A wrapper to combine multiple cells into a multi-layer cell.
  • DropoutWrapper – A wrapper to add dropout to input and/or output connections of a cell.
and the contributed RNN cells and wrappers:

Multi LSTM == Stacked LSTM
Multi Dimension LSTM == grid LSTM

No peephole connection or cell clipping
peephole connection: 那些gate會不會參考cell state

No comments:

Post a Comment