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