megengine.module.init.uniform_

uniform_(tensor, a=0.0, b=1.0)[源代码]

Fills the given tensor with random value sampled from uniform distribution \(\mathcal{U}(\text{a}, \text{b})\).

参数

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

  • a (float) – 采样区间的下界。

  • b (float) – 采样区间的上界。

返回类型

None

上一页

megengine.module.init.ones_

下一页

megengine.module.init.normal_