tensorflow中sum的用法是什么

小億
101
2024-03-30 12:25:57

在TensorFlow中，tf.reduce_sum函數(shù)用于計(jì)算張量中所有元素的總和。例如：

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。

0 踩