IndexCopy

Copies elements from source into input at indices.

Abstract Signature:

IndexCopy(input: Tensor, dim: int, index: Tensor, source: Tensor)

PyTorchJAX (Core)

PyTorch

API: torch.index_copy

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Plugin (scatter_indexer)