tensorflow::ops::UnsortedSegmentProd

#include <math_ops.h>

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

Summary

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

Этот оператор похож на оператор суммирования по сегментам без сортировки, который можно найти (здесь). Вместо вычисления суммы по сегментам он вычисляет произведение всех элементов, принадлежащих сегменту, так что:

\(output_i = {j...} data[j...]\) где произведение берётся по наборам j... таких, что segment_ids[j...] == i.

Например:

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

Если для данного идентификатора сегмента i нет элементов, то он выводит 1.

Если указанный идентификатор сегмента i отрицательный, соответствующее значение отбрасывается и не включается в результат.

Аргументы:

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

Возвращаемое значение:

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

Публичные атрибуты

Operation operation

::tensorflow::Output output

Публичные функции

 UnsortedSegmentProd(
  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