tensorflow::ops::UnsortedSegmentSum

#include <math_ops.h>

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

Краткое описание

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

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

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

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

Внимание! На CPU значения в segment_ids всегда проверяются на то, что они меньше num_segments; для индексов за пределами диапазона возникает ошибка. На GPU для индексов за пределами диапазона ошибка не возникает. На GPU индексы за пределами диапазона приводят к безопасному, но неопределённому поведению: такие индексы могут игнорироваться либо может возвращаться тензор, в первой размерности формы которого хранится 0, если num_segments равно 0.

c = [[1,2,3,4], [5,6,7,8], [4,3,2,1]] tf.math.unsorted_segment_sum(c, [0, 1, 0], num_segments=2).numpy() array([[5, 5, 5, 5], [5, 6, 7, 8]], dtype=int32)

Аргументы:

• scope: объект Scope
• segment_ids: тензор, форма которого является префиксом data.shape. Значения должны быть меньше num_segments.

Внимание! На CPU значения всегда проверяются на принадлежность диапазону, на GPU такая проверка не выполняется.

Возвращает:

• Output: Имеет ту же форму, что и data, за исключением первых 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