tf.reduce_sum
import numpy as np
import tensorflow as tf
x = np.array([[1,2,3],[4,5,6]])
sess = tf.Session()
sum0 = np.sum(x, axis=0)
sum1 = np.sum(x, axis=1)
sum2 = sess.run(tf.reduce_sum(x))
sum3 = sess.run(tf.reduce_sum(x, 0)) #0表示行
sum4 = sess.run(tf.reduce_sum(x, 1))#1表示列
sum5 = sess.run(tf.reduce_sum(x, 1, keep_dims=True))#keep_dims=True 表示不减少维度
sum6 = sess.run(tf.reduce_sum(x, [0, 1])) #行列加和
print (sum0)
print (sum1)
print (sum2)
print (sum3)
print (sum4)
print (sum5)
print (sum6)
sess.close()
输出: