MoveAxis

Move axes of an array to new positions.

Abstract Signature:

MoveAxis(a: Tensor, source: int | List[int], destination: int | List[int])

PyTorch

API: torch.moveaxis
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.numpy.moveaxis
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.moveaxis
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.ops.moveaxis
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.moveaxis
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.core.moveaxis
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: jax.numpy.moveaxis
Strategy: Direct Mapping
Official Docs ↗

PaxML / Praxis

API: jax.numpy.moveaxis
Strategy: Direct Mapping
Official Docs ↗