jax.random.permutation

目录

jax.random.permutation#

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

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

参数:

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

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

  • axis (int) – int, 可选。x 被随机打乱的轴。默认是 0。

  • independent (bool) – bool, 可选。如果设置为 True，则沿给定轴的每个单独向量将独立地进行洗牌。默认是 False。

返回:

x 的打乱版本或数组范围

返回类型:

Array

目录