Movedim

Moves the dimension(s) of input at the position(s) in source to the position(s) in destination.

Abstract Signature:

Movedim(input: Tensor, source: int, destination: int)

PyTorch

API: torch.movedim
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 ↗

Apple MLX

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