tf.math.reduce_sum():
有六个参数:
input_tensor, 顾名思义,输入的tensor,numeric型的数据
axis=None, 指定计算轴
keepdims=None, 保持维度
name=None, 为操作命名
reduction_indices=None, 旧版本中的axis,避免使用
keep_dims=None,不推荐使用,功能同keepdims
函数定义为:按照axis所给出的轴的方向(0为纵向1为横向)对tensor求和,若未指定axis则计算所有元素和并返回单个元素(相当于axis=[0,1])。
如果keepdim=True则减去的维度会保持长度为1。
返回类型与输入相同,是一个降维的tensor。
例子:
| 1 1 1 |
| 1 1 1 |=x 即x=constant([[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, keepdims=True) //[[3][3]]
tf.reduce_sum(x,[0,1]) //6