tensorflow::ops::UnsortedSegmentSum

#include <math_ops.h>

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

Summary

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

Вычисляет тензор такой, что \(output[i] = \sum_{j...} data[j...]\), где сумма берется по кортежам j... таким, что segment_ids[j...] == i. В отличие от SegmentSum, segment_ids не обязательно должны быть отсортированы и не обязательно должны охватывать все значения в полном диапазоне допустимых значений.

Если сумма пуста для данного идентификатора сегмента i, output[i] = 0. Если заданный идентификатор сегмента i отрицательный, значение отбрасывается и не будет добавлено к сумме сегмента.

num_segments должно быть равно количеству различных идентификаторов сегментов.

c = tf.constant([[1,2,3,4], [5,6,7,8], [4,3,2,1]])
tf.unsorted_segment_sum(c, tf.constant([0, 1, 0]), num_segments=2)
# ==> [[ 5,  5, 5, 5],
#       [5,  6, 7, 8]]

Аргументы:

• scope: Объект Scope
• segment_ids: Тензор, форма которого является префиксом data.shape.

Возвращает:

• Output: Имеет ту же форму, что и данные, за исключением первых segment_ids.rank измерений, которые заменяются одним измерением размером num_segments.

Общедоступные атрибуты

Operation operation

::tensorflow::Output output

Общедоступные функции

 UnsortedSegmentSum(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input data,
  ::tensorflow::Input segment_ids,
  ::tensorflow::Input num_segments
)

::tensorflow::Node * node() const 

operator::tensorflow::Input() const 

operator::tensorflow::Output() const