tf.reduce_sum()

Posted 笨拙的忍者

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了tf.reduce_sum()相关的知识,希望对你有一定的参考价值。

#axis 表示在哪个维度进行sum操作,不写代表所有维
#keep_dims 是否保留原始数据维度
reduce_sum(
    input_tensor,
    axis=None,
    keep_dims=False,
    name=None,
    reduction_indices=None
)

官方例子:

# ‘x‘ is [[1, 1, 1]
#         [1, 1, 1]]
tf.reduce_sum(x) ==> 6
tf.reduce_sum(x, 0) ==> [2, 2, 2]
tf.reduce_sum(x, 1) ==> [3, 3]
tf.reduce_sum(x, 1, keep_dims=True) ==> [[3], [3]]
tf.reduce_sum(x, [0, 1]) ==> 6

 

以上是关于tf.reduce_sum()的主要内容,如果未能解决你的问题,请参考以下文章

tf.reduce_sum函数

tf.reduce_sum()函数

TFboy养成记 tf.cast,tf.argmax,tf.reduce_sum

GPU 上的 tf.reduce_sum 结合占位符作为输入形状失败

Tensorflow - tf常用函数使用(持续更新中)

tensorflow学习