megengine.random package

megengine.random.distribution

megengine.random.distribution.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 ]]
megengine.random.distribution.uniform(low=0, high=1, size=None)[源代码]

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

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

  • low (float) – 范围下界

  • high (float) – 范围上界

返回类型

Tensor

返回

输出张量

示例代码:

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]]

megengine.random.rng

megengine.random.rng.seed(seed)[源代码]