tensorflow::ops::UnsortedSegmentProd

#include <math_ops.h>

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

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

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

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

\(output_i = \prod_{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