megengine.random.normal

normal(mean=0, std=1, size=None)

服从高斯分布的随机变量 \(N(\mu, \sigma)\)

参数
  • mean (float) – 分布的均值或期望。默认值:0

  • std (float) – 分布的标准差(方差= \(\sigma ^ 2\) )。默认值:1

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

返回

输出张量

实际案例

import megengine as mge
import megengine.random as rand

x = rand.normal(mean=0, std=1, size=(2, 2))
print(x.numpy())

输出:

[[-1.4010863  -0.9874344 ]
 [ 0.56373274  0.79656655]]