RandPerm

Returns a random permutation of integers from 0 to n - 1.

Abstract Signature:

RandPerm(n: int)

PyTorch

API: torch.randperm
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.random.permutation
Strategy: Plugin (inject_prng)

NumPy

API: np.random.permutation
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.numpy.random.permutation
Strategy: Macro 'keras.ops.convert_to_tensor(np.random.permutation({n}))'

TensorFlow

API: tf.random.shuffle
Strategy: Macro 'tf.random.shuffle(tf.range({n}))'

Apple MLX

API: mlx.core.random.permutation
Strategy: Macro 'mlx.core.random.permutation(mlx.core.arange({n}))'

Flax NNX

API: jax.random.permutation
Strategy: Plugin (inject_prng)