jax.numpy.permute_dims

Contents

jax.numpy.permute_dims#

jax.numpy.permute_dims(a, /, axes)[source]#

Parameters:

a (Union[Array, ndarray, bool_, number, bool, int, float, complex]) –
axes (tuple[int, ...]) –

Return type: