cyjax.util.shuffle_axis#

cyjax.util.shuffle_axis(key, x, axis)#

Shuffle indices of one axis.

Parameters:

  • key (Union[PRNGKeyArray, Array]) – PRNG random key.

  • x (Union[Array, ndarray, bool_, number]) – Array of values.

  • axis (int) – Axis to shuffle.

Return type:

Union[Array, ndarray, bool_, number]

Returns:

New array with values shuffled along given axis.