IndexReduce =========== Reduce values into tensor (e.g. mean, max). **Abstract Signature:** ``IndexReduce(input: Tensor, dim: int, index: Tensor, source: Tensor, reduce: str)`` .. raw:: html

PyTorch

API: torch.index_reduce
Strategy: Direct Mapping

JAX (Core)

API:
Strategy: Custom / Partial