ScatterReduce

Reduces values into a tensor at indices specified by index using a reduction operation.

Abstract Signature:

ScatterReduce(input: Tensor, dim: int, index: Tensor, src: Tensor, reduce: str, include_self: bool = True)

PyTorchJAX (Core)KerasTensorFlow

PyTorch

API: torch.scatter_reduce

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Plugin (scatter_indexer)

Keras

API: —

Strategy: Custom / Partial

TensorFlow

API: —

Strategy: Custom / Partial