KaimingUniform
class paddle.nn.initializer. KaimingUniform ( fan_in=None ) [源代码]
该接口实现Kaiming均匀分布方式的权重初始化
该接口为权重初始化函数,方法来自Kaiming He,Xiangyu Zhang,Shaoqing Ren 和 Jian Sun所写的论文: Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classification 。这是一个鲁棒性特别强的初始化方法,并且适应了非线性激活函数(rectifier nonlinearities)。
在均匀分布中,范围为[-x,x],其中:
参数:
- fan_in (float16|float32) - Kaiming Uniform Initializer的fan_in。如果为None,fan_in沿伸自变量,多设置为None
返回:对象
注解
在大多数情况下推荐设置fan_in为None
代码示例:
import paddle
import paddle.nn as nn
linear = nn.Linear(2, 4, weight_attr=nn.initializer.KaimingUniform())
data = paddle.rand([30, 10, 2], dtype='float32')
res = linear(data)