tf.reduce_sum()用法介绍

tf.reduce_sum()作用是按一定方式计算张量中元素之和
input_tensor为待处理张量；
axis指定按哪个维度进行加和，默认将所有元素进行加和；
keepdims默认为False，表示不维持原来张量的维度，反之维持原张量维度；
name用于定义该操作名字。
示例：
有如下张量，形状为(2,3,4)的三维张量：

分别按照axis=0,1，2进行加和的结果如下：

可以看到，原张量是三维的，指定axis进行一次tf.reduce_sum()后变成了二维张量。如果设置keepdims=True，则结果如下，可以看到计算完加和后的张量依然是三维的，其中指定axis的那个维度元素个数为1，其他维度形状不变。