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()的主要内容,如果未能解决你的问题,请参考以下文章
TFboy养成记 tf.cast,tf.argmax,tf.reduce_sum