TensorFlow RNN 单元类
tf.nn.rnn_cell
模块包含以下用于在 TensorFlow 中创建不同类型单元的类:
类 | 描述 |
---|---|
BasicRNNCell | 提供 RNN 单元的实现 |
BasicLSTMCell | 提供 LSTM RNN 单元的实现,基于 http://arxiv.org/abs/1409.2329 |
LSTMCell | 提供 LSTM RNN 单元,基于 http://deeplearning.cs.cmu.edu/pdfs/Hochreiter97_lstm.pdf 和 https:// research.google.com/pubs/archive/43905.pdf |
GRUCell | 提供 GRU RNN 单元,基于 http://arxiv.org/abs/1406.1078 |
MultiRNNCell | 提供由连续连接的多个简单单元组成的 RNN 单元 |
tf.contrib.rnn
模块提供以下额外的类用于在 TensorFlow 中创建不同类型的单元:
类 | 描述 |
---|---|
LSTMBlockCell | 提供块 LSTM RNN 单元,基于 http://arxiv.org/abs/1409.2329 |
LSTMBlockFusedCell | 提供块融合 LSTM RNN 单元,基于 http://arxiv.org/abs/1409.2329 |
GLSTMCell | 根据 https://arxiv.org/abs/1703.10722 提供组 LSTM 单元 |
GridLSTMCell | 提供网格 LSTM RNN 小区,基于 http://arxiv.org/abs/1507.01526 |
GRUBlockCell | 提供块 GRU RNN 单元,基于 http://arxiv.org/abs/1406.1078 |
BidirectionalGridLSTMCell | 仅在频率上而不是在时间上提供双向网格 LSTM |
NASCell | 提供神经架构搜索 RNN 单元,基于 https://arxiv.org/abs/1611.01578 |
UGRNNCell | 提供更新门 RNN 信元,基于 https://arxiv.org/abs/1611.09913 |