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) – int，可選。 x 沿其洗牌的軸。預設值為 0。

• independent (bool) – bool，可選。如果設定為 True，則沿給定軸的每個單獨向量都會獨立洗牌。預設值為 False。

返回:

x 的洗牌版本或陣列範圍

返回類型:

Array