megengine.random.normal

normal(mean=0, std=1, size=None)[源代码]

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

参数
  • size (Optional[Iterable[int]]) – 输出张量的形状

  • mean (float) – 分布的期望(平均数)

  • std (float) – 分布的标准差 (variance = \(\sigma ^ 2\))

返回类型

Tensor

返回

输出张量

例如:

import megengine as mge
import megengine.random as rand

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

输出:

[[-0.20235455 -0.6959438 ]
 [-1.4939808  -1.5824696 ]]