Roll

Roll the tensor along the given dimension(s).

Abstract Signature:

Roll(input: Tensor, shifts: Union[int, Tuple[int]], dims: Union[int, Tuple[int]])

PyTorch

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

JAX (Core)

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

NumPy

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

Keras

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

TensorFlow

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

Apple MLX

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

Flax NNX

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

PaxML / Praxis

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