megengine.module.init.xavier_uniform_

xavier_uniform_(tensor, gain=1.0)[源代码]

使用从 U(a,a) 中随机采样值填入 tensor 。其中,

a=gain×6fan_in+fan_out

又称Glorot初始化。详细的说明可以参考 “Understanding the difficulty of training deep feedforward neural networks”

参数
  • tensor (Tensor) – 待初始化的n维张量。

  • gain (float) – a 的比例因子。

返回类型

None