jax.random.permutation

内容

jax.random.permutation#

jax.random.permutation(key, x, axis=0, independent=False)[source]#

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

参数：

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

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

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

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

返回：

x 或数组范围的打乱版本

返回类型：

数组

内容