jax.random.permutation

jax.random.permutation(key, x, axis=0, independent=False)

返回一个随机排列的数组或范围。

参数:

• key (ArrayLike) – 用作随机密钥的 PRNG 密钥。

• x (int | ArrayLike) – 整数或数组。如果 x 是一个整数，则随机打乱 np.arange(x)。如果 x 是一个数组，则随机打乱其元素。

• axis (int) – 整数，可选。x 沿其打乱的轴。默认为 0。

• independent (bool) – 布尔值，可选。如果设置为 True，则沿给定轴的每个单独向量将独立打乱。默认为 False。

返回:

x 的打乱版本或数组范围

返回类型:

数组