在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。