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

上一页

megengine.module.init.calculate_correct_fan

下一页

megengine.module.init.xavier_normal_