tf.raw_ops.UnsortedSegmentProd

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

tf.raw_ops.UnsortedSegmentProd(
    data, segment_ids, num_segments, name=None
)

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

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

\(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 отрицательный, соответствующее значение отбрасывается и не включается в результат.