如何理解tf.contrib.rnn.BasicLSTMCell(num_units)中的num_units

基本的LSTM细胞单元在TensorFlow中声明为:

tf.contrib.rnn.BasicLSTMCell(num_units)

这里的num_units指的是LSTM单元中的单位数。

num_units也可以解释为前馈神经网络隐藏层的类比。前馈神经网络隐层中的节点。

num_units数目等于LSTM网络每个时间步长的LSTM单元的数量。以下图片应该可以帮助你理解:

5ca3b38110cd48ffb57cfe4a3afae787

 

有时用 lstm_size 表示 :这是 LSTM 网络的隐藏层中的单元数量,LSTM 细胞中实际上有四个不同的网络层,有三个 sigma 层和一个 TNH 层,这是每一层中的单元数。如果你将它设置成 256,那么这四层中的每一层就有 256 个单元。

num_units中的每一个:都可以将它看作LSTM网络中的一个标准的LSTM单元。

7bcc403a610f45a5bd4dcbbec7ed650e

 

参见:https://www.sohu.com/a/191568823_642762