Функция reduce_prod

Функция reduce_prod вычисляет произведение элементов тензора вдоль заданных осей. Первым параметром функция принимает тензор, вторым - ось или список осей для редукции. Третьим параметром можно указать логическое значение keepdims, которое сохраняет размерность исходного тензора.

Синтаксис

tf.reduce_prod(input_tensor, axis=None, keepdims=False)

Пример

Давайте вычислим произведение всех элементов одномерного тензора:

import tensorflow as tf

t = tf.constant([1, 2, 3, 4, 5])
res = tf.reduce_prod(t)
print(res)

Результат выполнения кода:

tf.Tensor(120, shape=(), dtype=int32)

Пример

Давайте вычислим произведение элементов двумерного тензора вдоль оси 0:

import tensorflow as tf

t = tf.constant([[1, 2, 3], [4, 5, 6]])
res = tf.reduce_prod(t, axis=0)
print(res)

Результат выполнения кода:

tf.Tensor([ 4 10 18], shape=(3,), dtype=int32)

Пример

Давайте вычислим произведение элементов двумерного тензора вдоль оси 1 с сохранением размерности:

import tensorflow as tf

t = tf.constant([[1, 2, 3], [4, 5, 6]])
res = tf.reduce_prod(t, axis=1, keepdims=True)
print(res)

Результат выполнения кода:

tf.Tensor(
[[  6]
 [120]], shape=(2, 1), dtype=int32)

Смотрите также

функцию reduce_sum,
которая вычисляет сумму элементов тензора
функцию reduce_mean,
которая вычисляет среднее значение элементов тензора
функцию reduce_max,
которая вычисляет максимальный элемент тензора
функцию cumprod,
которая вычисляет накопительное произведение элементов тензора