TensorFlow函数：tf.reduce_prod，sum,mean等其他详解

最新推荐文章于 2022-01-24 16:24:15 发布

原创

最新推荐文章于 2022-01-24 16:24:15 发布 · 4.8k 阅读

CC 4.0 BY-SA版权

现在tf1.7好像没有找到这个函数，但是却可以用，哎，谷歌API做的有点不专业啊。我翻了以前的文档里面有关于它的介绍：

tf.reduce_prod 函数

reduce_prod(
    input_tensor,
    axis=None,
    keep_dims=False,
    name=None,
    reduction_indices=None
)

此函数计算一个张量的各个维度上元素的乘积。

函数中的input_tensor是按照axis中已经给定的维度来减少的；除非 keep_dims 是true，否则张量的秩将在axis的每个条目中减少1；如果keep_dims为true，则减小的维度将保留为长度1。

如果axis没有条目，则缩小所有维度，并返回具有单个元素的张量。

参数：

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