Tensorflow 函数说明tf.reduce_sum

 tf.reduce_sum(input_tensor, axis=None, keep_dims=False, name=None, reduction_indices=None)

函数意义:将矩阵的元素相加,可以按列,也可以按行,或者先按列后按行

程序例程如下:

def Test_Reduce_Sum():
    x=[[1,1,1],[2,2,2]]
    input=tf.constant(x,tf.int32)
    #矩阵整体求和
    sum= tf.reduce_sum(input)
    # 矩阵按列求和
    sum1=tf.reduce_sum(input,reduction_indices=0)# 等价 tf.reduce_sum(input,axis=0)
    # 矩阵按行求和
    sum2 = tf.reduce_sum(input, reduction_indices=1)# 等价 tf.reduce_sum(input,axis=1)
    # 矩阵按列求和之后按行求和
    sum3 = tf.reduce_sum(input, reduction_indices=[0,1])


结果如下:

9
[3 3 3]
[3 6]
9

这样的函数在tensorflow中还有很多: 

tf.reduce_prod #沿维度相乘

tf.reduce_min #沿维度找最小

tf.reduce_max #沿维度找最大

tf.reduce_mean #沿维度求平均

tf.reduce_all#沿维度与操作

tf.reduce_any#沿维度或操作

具体可以参考点击打开链接



你可能感兴趣的:(Tensorflow)