Narrow

Returns a new tensor that is a narrowed version of input tensor.

Abstract Signature:

Narrow(input: Tensor, dim: int, start: int, length: int)

PyTorchJAX (Core)NumPyApple MLX

PyTorch

API: torch.narrow

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.lax.dynamic_slice_in_dim

Strategy: Direct Mapping

Official Docs ↗

NumPy

API: —

Strategy: Plugin (slice_generator)

Apple MLX

API: —

Strategy: Plugin (slice_generator)