mlx.core.random.multivariate_normal

多元正态分布(均值： 数组, 协方差： 数组, 形状： 序列[int] = [], 数据类型： 数据类型 | 无 = 浮点32, 键： 数组 | 无 = 无, 流： None | Stream | Device = 无) → 数组

给定均值和协方差生成联合正态随机样本。

矩阵 cov 必须是半正定的。如果不是，行为是未定义的。唯一支持的 dtype 是 float32。

Parameters：

• mean (array) – 形状为 (..., n) 的数组，表示分布的均值。

• 协方差 (数组) – 形状为 (..., n, n) 的数组，表示分布的协方差矩阵。批次形状 ... 必须与 均值 的批次形状兼容。

• shape (list(int), optional) – 输出形状必须与 mean.shape[:-1] 和 cov.shape[:-2] 广播兼容。 如果为空，结果形状由 mean 和 cov 的批次形状广播决定。默认值：[]。

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

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

Returns：

随机值的输出数组。

Return type：

数组