tensorflow::ops::UnsortedSegmentProd

#include <math_ops.h>

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

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

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

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

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