Python tf.math.reduce_prod用法及代码示例

跨张量的维度计算元素的tf.math.multiply。

用法

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

参数

input_tensor 要减少的张量。应该是数字类型。
axis 要减小的尺寸。如果None(默认)，减少所有维度。必须在 [-rank(input_tensor), rank(input_tensor)) 范围内。
keepdims 如果为真，则保留长度为 1 的缩减维度。
name 操作的名称(可选)。

减少的张量。

这是按元素tf.math.multiply op 的归约操作。

沿 axis 中给定的尺寸减少 input_tensor 。除非 keepdims 为真，否则对于 axis 中的每个条目，张量的秩都会减少 1。如果 keepdims 为真，则保留缩减后的维度，长度为 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)>

numpy 兼容性

相当于 np.prod

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.math.reduce_prod。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。