megengine.random.uniform

uniform(low=0, high=1, size=None)[源代码]

服从均匀分布 $U(0, 1)$ 的随机变量

参数
  • size – 输出张量的形状

  • low – 范围下界

  • high – 范围上界

返回

输出张量

例如:

import megengine as mge
import megengine.random as rand

x = rand.uniform(size=(2, 2))
print(x.numpy())

输出:

[[0.76901674 0.70496535]
 [0.09365904 0.62957656]]