- jax.numpy.moveaxis(a, source, destination)#
Move axes of an array to new positions.
LAX-backend implementation of
The JAX version of this function may in some cases return a copy rather than a view of the input.
Original docstring below.
Other axes remain in their original order.
New in version 1.11.0.
result – Array with moved axes. This array is a view of the input array.
- Return type: