tf.reduce_sum降维函数

作者：Eliiiiiiiii

链接：https://www.zhihu.com/question/51325408/answer/238082462

来源：知乎

著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

计算方法：

规律：

对于k维的，tf.reduce_sum(x, axis=k-1)的结果是对最里面一维所有元素进行求和。tf.reduce_sum(x, axis=k-2)是对倒数第二层里的向量对应的元素进行求和。tf.reduce_sum(x, axis=k-3)把倒数第三层的每个向量对应元素相加。

用tf.reduce_sum里的例子进行检验：

# 'x' is [[1, 1, 1]

#        [1, 1, 1]]

tf.reduce_sum(x, 0) ==> [2, 2, 2] # 倒数第二层对应元素相加

tf.reduce_sum(x, 1) ==> [3, 3] #最里面一层求和