在TensorFlow中,tf.reduce_sum
函数用于计算张量中所有元素的总和。例如:
import tensorflow as tf
x = tf.constant([[1, 2, 3], [4, 5, 6]])
sum_x = tf.reduce_sum(x)
with tf.Session() as sess:
result = sess.run(sum_x)
print(result)
这将输出:
21
上面的代码将张量x
中的所有元素相加,得到总和为21。