LSTM需要学习的参数个数

本文介绍了如何计算LSTM网络中需要学习的参数数量。通过一个实例,详细解析了输入28维向量映射到128维,以及处理13*5大小图片时,隐藏层神经元为10的LSTM单元的参数计算过程,最终得出LSTM所需学习的参数总数为640个。
摘要由CSDN通过智能技术生成

 结论先行:

lstm所需学习的参数 = 4 * [(  input_size + hidden_num )*  hidden_num + hidden_num ]

我们知道输入一个28维的向量,会映射到128维来进行表示,

输出ht和记忆ct维度相同。

下面我们假设有13*5大小的图片,放进LSTM,隐含层神经元数量为10

 

ht-1 是上一时刻的输出,维数与隐含层神经元数量一致:1*10

xt为1*5 所以[ ht-1 , xt ] 的维数是1*15

又因为激活函数不改变位数维数,所以我们不必考虑

Wf  bf 为该层的参数,Wf  与 [ ht-1 , xt ] 相乘得到 1*10的向量

所以可以得出 Wf 的维数是 15*10    即:Wf  × [ ht-1 , xt ] = 15*10  × 1*15  = 1*10

所以参数个数为Wf  + bf  = 15*10 + 10 = 160个

剩下的操作与这个遗忘门的操作在参数数量上相同

故 LSTM的所需学习参数个数为 160*4 = 640个

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值