tf.reduce_sum( ) 函数是如何使用的

tf.reduce_sum( ) 函数用法

def reduce_sum_v1(input_tensor,
                  axis=None,	
                  keepdims=None,
                  name=None,
                  reduction_indices=None,
                  keep_dims=None):

reduce_sum( ) 是求和函数

reduce_sum() 就是求和，由于求和的对象是tensor，所以是沿着tensor的某些维度求和。reduction_indices是指沿tensor的哪些维度求和，下面以一个例子形容维度求和的具体操作：

例子

tensor = tf.constant([[1,1,1],[1,1,1]])

当axis不给定时

 tf.reduce_sum(input_tensor = tensor) 
return 6 

当axis=0时

 tf.reduce_sum(input_tensor = tensor，axis=0) 
return [2,2,2]

当axis=1时

 tf.reduce_sum(input_tensor = tensor，axis=0) 
return [3,3]

当keepdims=True时

 tf.reduce_sum(input_tensor = tensor，axis=0，keepdims=True) 
return [[3],[3]]