IndexAdd

Accumulate values into tensor at indices.

Abstract Signature:

IndexAdd(input: Tensor, dim: int, index: Tensor, source: Tensor, alpha: number = 1)

PyTorchJAX (Core)Keras

PyTorch

API: torch.index_add

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.numpy.ndarray.at

Strategy: Plugin (scatter_indexer)

Keras

API: keras.ops.scatter_add

Strategy: Direct Mapping

Official Docs ↗