Rollaxis

Roll the specified axis to a given position.

Abstract Signature:

Rollaxis(a: Tensor, axis: int, start: int = 0)

PyTorchJAX (Core)NumPy

PyTorch

API: —

Strategy: Custom / Partial

JAX (Core)

API: jax.numpy.rollaxis

Strategy: Direct Mapping

Official Docs ↗

NumPy

API: numpy.rollaxis

Strategy: Direct Mapping

Official Docs ↗