Computes a tensor such that
\(output_i = \sum_j data_j\) where sum is over j such
that segment_ids[j] == i.
If the sum is empty for a given segment ID i, output[i] = 0.
Note that this op is currently only supported with jit_compile=True.
Args
data
A Tensor. Must be one of the following types: float32, float64, int32, uint8, int16, int8, complex64, int64, qint8, quint8, qint32, bfloat16, qint16, quint16, uint16, complex128, half, uint32, uint64.
segment_ids
A Tensor. Must be one of the following types: int32, int64.
A 1-D tensor whose size is equal to the size of data's
first dimension. Values should be sorted and can be repeated.
The values must be less than num_segments.
num_segments
A Tensor. Must be one of the following types: int32, int64.
[null,null,["Last updated 2024-04-26 UTC."],[],[],null,["# tf.raw_ops.SegmentSumV2\n\nComputes the sum along segments of a tensor.\n\n#### View aliases\n\n\n**Compat aliases for migration**\n\nSee\n[Migration guide](https://www.tensorflow.org/guide/migrate) for\nmore details.\n\n[`tf.compat.v1.raw_ops.SegmentSumV2`](https://www.tensorflow.org/api_docs/python/tf/raw_ops/SegmentSumV2)\n\n\u003cbr /\u003e\n\n tf.raw_ops.SegmentSumV2(\n data, segment_ids, num_segments, name=None\n )\n\nRead\n[the section on segmentation](https://tensorflow.org/api_docs/python/tf/math#Segmentation)\nfor an explanation of segments.\n\nComputes a tensor such that\n\\\\(output_i = \\\\sum_j data_j\\\\) where sum is over `j` such\nthat `segment_ids[j] == i`.\n\nIf the sum is empty for a given segment ID `i`, `output[i] = 0`.\n\nNote that this op is currently only supported with jit_compile=True.\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Args ---- ||\n|----------------|------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| `data` | A `Tensor`. Must be one of the following types: `float32`, `float64`, `int32`, `uint8`, `int16`, `int8`, `complex64`, `int64`, `qint8`, `quint8`, `qint32`, `bfloat16`, `qint16`, `quint16`, `uint16`, `complex128`, `half`, `uint32`, `uint64`. |\n| `segment_ids` | A `Tensor`. Must be one of the following types: `int32`, `int64`. A 1-D tensor whose size is equal to the size of `data`'s first dimension. Values should be sorted and can be repeated. The values must be less than `num_segments`. \u003cbr /\u003e | **Caution:** The values are always validated to be sorted on CPU, never validated on GPU. |\n| `num_segments` | A `Tensor`. Must be one of the following types: `int32`, `int64`. |\n| `name` | A name for the operation (optional). |\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Returns ------- ||\n|---|---|\n| A `Tensor`. Has the same type as `data`. ||\n\n\u003cbr /\u003e"]]
