这里num_units是指LSTM(或rnn)单元格中的单元数。
num_units可以被解释为来自前馈神经网络的隐藏层的类比。前馈神经网络的隐藏层中的节点数量等于网络的每个时间步长处的LSTM小区中的LSTM单元的num_units数量。以下图片应清除任何混淆 - 在此输入图像描述
(引自 https://jasdeep06.github.io/posts/Understanding-LSTM-in-Tensorflow-MNIST/
不,这不对。
num_units 指的是您的单元格可以表示的要素数量。在每个时间步,您给出一定大小的输入(您称之为“RNN应按顺序处理的元素数量”)。这就像你的神经网络的第0层。然后将此输入处理为具有大小的隐藏层 num_units 。这也是单元格输出的大小。
num_units
您调用N的是由输入张量的大小设置的。 num_units 是你的模型的超参数。它越大,模型的自由度就越大(更具描述性的特征)。