如何为架构示意图表示 LSTM 中的神经元数量?

数据挖掘 神经网络 lstm 可视化 rnn
2022-02-24 21:57:47

我正在尝试可视化神经网络示意图,并在http://alexlenail.me/NN-SVG/index.html找到了构建示意图的好工具。我编辑了 SVG 文件,将其中一个密集层更改为 LSTM 层,并将输入更改为时间序列而不是单个神经元。

在图像的底部有一些集合符号,详细说明了每层中有多少个神经元。我对集合表示法不太熟悉。我不太确定如何根据给定的集合符号来表示 LSTM 层的神经元数量?

它是一个具有 hidden_​​size=8 的奇异 LSTM 层,输入/输出大小显然只是前一层/下一层(5 和 8)。

同样,同样的问题也适用于初始时间序列输入;如何用集合符号表示 5 个时间序列(比如 100 点长)?

关于如何更好地表示架构的任何其他建议也将不胜感激!谢谢!

神经架构

1个回答

在循环单元中,您可以添加隐藏单元 - 通常,我们将其标记为:

Unit 1
  ...  
Unit n

n循环表示的维数在哪里。