megengine.module.init.uniform_

uniform_(tensor, a=0.0, b=1.0)

从均匀分布 \(\mathcal{U}(\text{a}, \text{b})\) 中采样随机值填充给定 tensor。

参数

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

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

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

返回类型

None

上一页

megengine.module.init.ones_

下一页

megengine.module.init.normal_