Warning

This page was created from a pull request.

jax.random.permutation

jax.random.permutation(key, x)

Permute elements of an array along its first axis or return a permuted range.

If x is a multi-dimensional array, it is only shuffled along its first index.

Args:n

key: a PRNGKey used as the random key. x: the array or integer range to be shuffled.

Returns

A shuffled version of x or array range