nn.GRUCell过程思考

本文探讨PyTorch官方文档中的GRUCell,分析其数学公式,揭示内部结构。GRUCell包含input-hidden和hidden-hidden两部分,简化了LSTM的网络结构,通过update门控制长期和短期特征的学习,降低梯度消失问题。GRU是RNN的一种变体,理解GRU需要对RNN有基础认知。
摘要由CSDN通过智能技术生成

记录一下看pytorch官方文档的思考

首先是公式

 看起来像是在更新h,新的h等于n和旧h的组合,它们的组合系数是z;就像调酒,新h等于来z个旧h,剩下都是(1-z)个n;

在使用时先定义这个网络的大小,也就是输入特征的个数和隐藏状态的个数

然后调用的时候要送入输入特征x和隐藏状态h,这俩都是张量

所有的W都是参数,前三个公式每个公式有两个W

从这里可以看出,GRUCell的内部好像还有一层隐藏层,结构大致是input-hidden,hidden-hidden(output)。

第一个公式求r把输入乘权重+隐藏状态个数乘权重+两个偏移量,统一sigmoid激活一下;

第二个公式求z做法一致,就是各种参数变了

第三 个公式求n,用到r的结果,基本就是权重一乘再相加起来,tanh激活一下

第四个公式更新h,新h就是调酒组合

从公式看来我的理解是第二个公式相当于input-hidden,第三个公式是hidden-hidden;

至于这个GRUCell,就是GRU的单个单元;GRU算法出自

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值