这个函数一般使用如下:
import tensorflow as tf
x = tf.constant([[1,1,1],[2,2,2]])
y1 = tf.reduce_sum(x,0)----->[3,3,3]
y2 = tf.reduce_sum(x,1)----->[3,6]
y3 = tf.reduce_sum(x,1,keep_dims=True)----->[[3],[6]]
从函数名字上理解,相当于一种减少维度的计算函数。
那么上面这个函数的里边参数:0\1代表什么意思?看下图:
当然,还有下面这种用法:
y4 = reduce_sum(x)
对于[[1,1,1],1,1,1](就是上图这种形式),使用y4=reduce_sum(x)=6