RollAxis ======== Roll the specified axis backwards, until it lies in a given position. **Abstract Signature:** ``RollAxis(a: Array, axis: int, start: int = 0)`` .. raw:: html

PyTorch

API:
Strategy: Plugin (rollaxis_to_moveaxis)

JAX (Core)

API: jax.numpy.rollaxis
Strategy: Direct Mapping

NumPy

API: numpy.rollaxis
Strategy: Direct Mapping

Flax NNX

API: jax.numpy.rollaxis
Strategy: Direct Mapping

PaxML / Praxis

API: jax.numpy.rollaxis
Strategy: Direct Mapping