LSTM 单元:细胞状态维度

数据挖掘 喀拉斯 rnn lstm
2022-03-14 15:40:37

当我们拟合 LSTM 模型时,每个 LSTM 都有一个包含我们想要的信息的单元状态。我想知道细胞状态的维度是多少(即细胞状态本身应该是一个向量,那么向量有多大?)?

在此处输入图像描述 (来自http://colah.github.io/posts/2015-08-Understanding-LSTMs/

即在上述架构中,我们如何确定矩阵的维数Wc

如果我使用 Keras 包,如何找到每个 LSTM 单元的单元状态维度?是可以调整的吗?

谢谢!

2个回答

在 Keras 中,LSTM 中的第一个参数给出了细胞状态的维度。

注意下一个隐藏状态ht通过将逐点标量操作应用于下一个单元状态来产生Ct. (精确公式ht=ottanh(Ct)可以在您链接到的博客文章中找到。)因此,单元状态和隐藏状态具有相同的维度。