IndexReduce

Reduce values into tensor (e.g. mean, max).

Abstract Signature:

IndexReduce(input: Tensor, dim: int, index: Tensor, source: Tensor, reduce: str)

PyTorchJAX (Core)

PyTorch

API: torch.index_reduce

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Custom / Partial