TensorFlow api: reduce_sum

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

猜你喜欢

转载自blog.csdn.net/cyr429/article/details/89242923
今日推荐