NarrowCopy

Same as narrow except this returns a copy rather than shared storage.

Abstract Signature:

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

PyTorch

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

JAX (Core)

API: jax.lax.dynamic_slice_in_dim
Strategy: Direct Mapping
Official Docs ↗