jax.numpy.moveaxis#
- jax.numpy.moveaxis(a, source, destination)[source]#
Move axes of an array to new positions.
LAX-backend implementation of
numpy.moveaxis()
.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.
- Parameters:
- Returns:
result – Array with moved axes. This array is a view of the input array.
- Return type:
np.ndarray