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 |