tf.keras.ops.segment_sum

Вычисляет сумму отрезков в тензоре.

tf.keras.ops.segment_sum(
    data, segment_ids, num_segments=None, sorted=False
)

Аргументы
`data`	Входной тензор.
`segment_ids`	Одномерный тензор, содержащий индексы отрезков для каждого элемента в `data`.
`num_segments`	Целое число, представляющее общее количество отрезков. Если не указано, оно выводится из максимального значения в `segment_ids`.
`sorted`	Булево значение, указывающее, отсортирован ли `segment_ids`. По умолчанию `False`.

Возвращаемое значение
Тензор, содержащий сумму отрезков, где каждый элемент представляет сумму соответствующего отрезка в `data`.

Пример:

data = keras.ops.convert_to_tensor([1, 2, 10, 20, 100, 200])
segment_ids = keras.ops.convert_to_tensor([0, 0, 1, 1, 2, 2])
num_segments = 3
keras.ops.segment_sum(data, segment_ids,num_segments)
array([3, 30, 300], dtype=int32)

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/ops/segment_sum