megengine.random.uniform

uniform(low=0, high=1, size=None)

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

参数
  • low (float) – 下界。默认值:0

  • high (float) – 上界。默认值:1

  • size (Optional[Iterable[int]]) – 输出向量的大小。默认值:None

返回

输出张量

实际案例

>>> import megengine.random as rand
>>> x = rand.uniform(size=(2, 2))
>>> x.numpy()   
array([[0.28603864, 0.3156649 ],
       [0.42066026, 0.9805052 ]], dtype=float32)