TensorFlow reduce_sum

最新推荐文章于 2022-03-09 10:43:58 发布

二分掌柜的

最新推荐文章于 2022-03-09 10:43:58 发布

阅读量3.6k

点赞数

CC 4.0 BY-SA版权

分类专栏： # TensorFlow 文章标签： TensorFlow

本文链接：https://blog.csdn.net/flyfish1986/article/details/54646236

TensorFlow 专栏收录该内容

27 篇文章

订阅专栏

本文通过一个具体的示例代码介绍了如何使用 TensorFlow 中的 reduce_sum 函数来对多维数组进行求和操作，并展示了不同参数设置下 reduce_sum 的行为表现。

TensorFlow reduce_sum

flyfish

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))
sum4 = sess.run(tf.reduce_sum(x, 1))
sum5 = sess.run(tf.reduce_sum(x, 1, 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()