tf.math.unsorted_segment_prod

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

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