IndexPut

Puts values into tensor at indices (tuple).

Abstract Signature:

IndexPut(input: Tensor, indices: Tuple[Tensor], values: Tensor, accumulate: bool = False)

PyTorchJAX (Core)

PyTorch

API: torch.index_put

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Plugin (scatter_indexer)