nn.Parameter 详解 (将 nn.Parameter 用于 nn.init.kaiming_normal 初始化)

Parameters 是 Variable 的子类。Paramenters和Modules一起使用的时候会有一些特殊的属性,即:当Paramenters赋值给Module的属性的时候,它会自动的被加到 Module的 参数列表中(即:会出现在 parameters() 迭代器中)。
简单来说就是将一个不可训练的 tensor 转换成可以训练的类型 parameter 并将这个 parameter 绑定到这个module 里面,在BP过程中就可以实现对其优化。例如将parameter 作为模型初始化。

att_weights = nn.Parameter(torch.Tensor(1, 256),requires_grad=True)
nn.init.kaiming_normal(att_weights, mode=‘fan_out’)

参数说明:
data (Tensor) – parameter tensor.
requires_grad (bool, optional) – 默认为True,在BP的过程中会对其求微分。

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值