IndexPut

In-place version of IndexPut.

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 (unroll_inplace_ops)