LSTM 中的 NumHiddenUnits

Question

当我们在 matlab 中创建一个 LSTM 层时，我们将 numHiddenUnits 指定为 layer = lstmLayer(numHiddenUnits)。我有两个问题。

(1) ：numHiddenUnits 是什么意思？

是否代表LSTM个细胞的数量？如果是，那么这些电池是串联还是并联（它们之间没有相互作用）。

不幸的是，没有办法可视化 (RNN) 网络。我了解一个 LSTM 单元的工作原理，但我不知道以下架构如何。

我的网络由这些层组成。

layers = [ ...
    sequenceInputLayer(numFeatures)
    lstmLayer(numHiddenUnits)
    fullyConnectedLayer(numResponses)
    regressionLayer];

(2): 这个架构是怎样的？

我的方法：我尝试画出它的草图，我认为它应该是这样的。

Answer 1

numHiddenUnits是LSTM隐藏状态的维数。例如，如果您设置 numHiddenUnits = 5 ，则 LSTM 输出是一个 5 维向量。所以，它并不代表 LSTM 单元的数量。
This tutorial 将帮助您更好地理解您的模型。您的模型是这样工作的：每次 LSTM 收到一个输入并对其进行处理，但直到最后一个时间步才会输出。在最后一个时间步，您的 LSTM 输出一个向量并将其发送到全连接层并为您提供回归值。如果只有你的最后一个 LSTM 单元将输出发送到全连接层，而不是所有的层，你的草图就可以了。

希望对您有所帮助

NumHiddenUnits in LSTM