jax.random.permutation

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

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

参数:

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

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

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

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

返回:

x 或数组范围的打乱版本

返回类型:

数组 (Array)