mlx.core.random.normal

目录

mlx.core.random.normal#

normal(shape: Sequence[int] = [], dtype: Dtype | None = float32, loc: float = 0.0, scale: float = 1.0, key: array | None = None, stream: None | Stream | Device = None) → array#

生成服从正态分布的随机数。

参数：

  • shape (list(int), 可选) – 输出的形状。默认值为 ()。

  • dtype (Dtype, 可选) – 输出的类型。默认值为 float32。

  • loc (float, 可选) – 分布的均值。默认值为 0.0。

  • scale (float, 可选) – 分布的标准差。默认值为 1.0。

  • key (array, 可选) – 一个 PRNG 键。默认值：None。

返回值：

随机值的输出数组。

返回类型：

array

目录