тензорный поток:: опс:: НесортированнаяСегментСумма

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

Краткое содержание

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

Вычисляет тензор такой, что \(output[i] = {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]]

Аргументы:

• область: объект области.
• сегмент_идс: тензор, форма которого является префиксом 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