megengine.module.init.xavier_uniform_

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

使用从 \(\mathcal{U}(-a, a)\) 中随机采样值填入 tensor 。其中,

\[a = \text{gain} \times \sqrt{\frac{6}{\text{fan_in} + \text{fan_out}}}\]

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

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

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

返回类型

None