SegmentReduce

Performs a reduction along segments of a tensor.

Abstract Signature:

SegmentReduce(data: Tensor, reduce: str, lengths: Tensor, indices: Tensor, offsets: Tensor, axis: int = 0)

PyTorch

API: torch.segment_reduce
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.ops.segment_sum
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.ops.segment_sum
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.math.segment_sum
Strategy: Direct Mapping
Official Docs ↗