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.