SelectCopy

Slices the tensor, returning a copy.

Abstract Signature:

SelectCopy(input: Tensor, dim: int, index: int)

PyTorchJAX (Core)NumPy

PyTorch

API: torch.select_copy

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.lax.index_take

Strategy: Direct Mapping

Official Docs ↗

NumPy

API: numpy.take

Strategy: Direct Mapping

Official Docs ↗