Pytorch网络参数初始化方法总结torch.nn.init

本文总结了PyTorch中几种常用的网络参数初始化方法,包括均匀分布、正态分布、Xavier和Kaiming初始化。详细介绍了Xavier均匀和正态分布以及Kaiming均匀和正态分布的实现,特别指出它们在不同激活函数下的适用性。
摘要由CSDN通过智能技术生成

参考1:官网
参考2:博客

均匀分布初始化torch.nn.init.uniform_()

torch.nn.init.uniform_(tensor, a=0.0, b=1.0)

使输入的张量服从(a,b)的均匀分布并返回。
参数

  • tensor – n维张量
  • a –均匀分布的上界
  • b –均匀分布的下界

正态分布初始化torch.nn.init.normal_()

torch.nn.init.normal_(tensor, mean=0.0, std=1.0)

从给定的均值和标准差的正态分布N(mean,std2)中生成值,初始化张量。
参数

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值