tf.math.reduce_prod
| Просмотреть исходный код на GitHub |
Вычисляет произведение элементов по измерениям тензора.
tf.math.reduce_prod(
input_tensor, axis=None, keepdims=False, name=None
)
Это операция сокращения для элементной операции умножения.
Сокращает элементы по измерениям, указанным в axis. Если keepdims не имеет значения true, ранг тензора уменьшается на 1 для каждого элемента в axis. Если keepdims имеет значение true, сокращенные измерения сохраняются с длиной 1.
Если axis имеет значение None, все измерения сокращаются, и возвращается тензор с одним элементом.
Например:
x = tf.constant([[1., 2.], [3., 4.]]) tf.math.reduce_prod(x) <tf.Tensor: shape=(), dtype=float32, numpy=24.> tf.math.reduce_prod(x, 0) <tf.Tensor: shape=(2,), dtype=float32, numpy=array([3., 8.], dtype=float32)> tf.math.reduce_prod(x, 1) <tf.Tensor: shape=(2,), dtype=float32, numpy=array([2., 12.], dtype=float32)>
| Аргументы | |
|---|---|
input_tensor | Тензор для сокращения. Должен иметь числовой тип. |
axis | Измерения для сокращения. Если None (по умолчанию), сокращаются все измерения. Должно находиться в диапазоне [-rank(input_tensor), rank(input_tensor)). |
keepdims | Если true, сохраняет сокращенные измерения с длиной 1. |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
| Сокращенный тензор. |
Совместимость с numpy
Эквивалентно np.prod
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/math/reduce_prod