UnsortedSegmentProd =================== Computes the product along segments of a tensor. **Abstract Signature:** ``UnsortedSegmentProd(data: Tensor, segment_ids: Tensor, num_segments: int)`` .. raw:: html

JAX (Core)

API: jax.ops.segment_prod
Strategy: Direct Mapping

TensorFlow

API: tf.math.unsorted_segment_prod
Strategy: Direct Mapping

Flax NNX

API: jax.ops.segment_prod
Strategy: Direct Mapping

PaxML / Praxis

API: jax.ops.segment_prod
Strategy: Direct Mapping