mlx.core.random.permutation

permutation(x: int | array, axis: int = 0, key: array | None = None, stream: None | Stream | Device = None) → array

生成一个随机排列或对数组的条目进行随机重排。

参数：

• x (int 或 array, 可选) – 如果提供一个整数，则返回 mx.arange(x) 的随机排列。否则，将对 x 沿给定轴的条目进行随机重排。

• axis (int, 可选) – 进行重排的轴。默认值：0。

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

返回值：

生成的随机排列或随机重排的输入数组。

返回类型：

array