时间戳

首页 > 解决方案 > LSTM 中的 NumHiddenUnits

matlab - LSTM 中的 NumHiddenUnits

问题描述

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

(1) : 是什么意思numHiddenUnits？

它代表LSTM细胞的数量吗？如果是，那么这些电池是串联连接还是并联连接（它们之间没有交互）。

不幸的是，没有办法可视化（RNN）网络。我了解一个 LSTM 单元是如何工作的，但我不知道下面的架构是什么样的。

我的网络由这些层组成。

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

(2)：这个架构是什么样子的？

我的方法：我试着画出它，我认为它应该是这样的。 在此处输入图像描述

标签： matlabmachine-learningneural-networklstmrecurrent-neural-network

解决方案

  1. numHiddenUnits是 LSTM 隐藏状态的维数。例如，如果您设置numHiddenUnits = 5，则 LSTM 输出是一个 5 维向量。因此，它并不代表 LSTM 单元的数量。

  2. 本教程将帮助您更好地理解您的模型。您的模型是这样工作的：每次 LSTM 接收输入并对其进行处理，但直到最后一个时间步才输出。在最后一个时间步，您的 LSTM 输出一个向量并将其发送到全连接层，并为您提供回归值。如果只有你的最后一个 LSTM 单元将输出发送到全连接层，你的草图就可以了，而不是全部。

希望这可以帮助

推荐阅读