tf.raw_ops.SparseSegmentSum

Вычисляет сумму по разреженным сегментам тензора.

Псевдонимы

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительных сведений.

tf.compat.v1.raw_ops.SparseSegmentSum

tf.raw_ops.SparseSegmentSum(
    data, indices, segment_ids, sparse_gradient=False, name=None
)

Прочитайте раздел о сегментации для объяснения сегментов.

Подобно SegmentSum, но segment_ids может иметь ранг меньше, чем ранг первого измерения data, выбирая подмножество нулевого измерения, указанное через indices.

Например:

c = tf.constant([[1,2,3,4], [-1,-2,-3,-4], [5,6,7,8]])

# Select two rows, one segment.
tf.sparse_segment_sum(c, tf.constant([0, 1]), tf.constant([0, 0]))
# => [[0 0 0 0]]

# Select two rows, two segment.
tf.sparse_segment_sum(c, tf.constant([0, 1]), tf.constant([0, 1]))
# => [[ 1  2  3  4]
#     [-1 -2 -3 -4]]

# Select all rows, two segments.
tf.sparse_segment_sum(c, tf.constant([0, 1, 2]), tf.constant([0, 0, 1]))
# => [[0 0 0 0]
#     [5 6 7 8]]

# Which is equivalent to:
tf.segment_sum(c, tf.constant([0, 0, 1]))

Аргументы
`data`	А `Tensor`. Должен быть одним из следующих типов: `float32`, `float64`, `int32`, `uint8`, `int16`, `int8`, `int64`, `bfloat16`, `uint16`, `half`, `uint32`, `uint64`.
`indices`	А `Tensor`. Должен быть одним из следующих типов: `int32`, `int64`. Одномерный тензор. Имеет тот же ранг, что и `segment_ids`.
`segment_ids`	А `Tensor`. Должен быть одним из следующих типов: `int32`, `int64`. Одномерный тензор. Значения должны быть отсортированы и могут повторяться.
`sparse_gradient`	Необязательный `bool`. По умолчанию `False`.
`name`	Имя операции (необязательно).

Возвращаемое значение
А `Tensor`. Имеет тот же тип, что и `data`.

© 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/raw_ops/SparseSegmentSum