tf.reduce_sum(tensor, axis=-1)

静静_jingjing

于 2021-05-24 10:40:26 发布

阅读量1.8k

点赞数

分类专栏： python编程

本文链接：https://blog.csdn.net/qq_39321513/article/details/117219041

版权

TensorFlow reduce_sum 张量操作轴向求和会话运行

关键词由CSDN通过智能技术生成

python编程专栏收录该内容

36 篇文章

订阅专栏

一个很好的介绍的博客：彻底理解 tf.reduce_sum() - 简书 (jianshu.com)

代码举例：

sess = tf.Session()
tensor = [[[ 1 ,  2 ,  3 ,  4],
  [ 5  , 6,   7 ,  8],
  [ 9 ,  10, 11, 12]],
 [[ 13,  14 ,15 ,16],
  [ 17 , 18 ,19, 20],
  [ 21,  22 ,23 ,24]]]
import numpy as np
c= tf.reduce_sum(tensor, axis=-1)
print( sess.run(c))

结果：

[[10 26 42]
[58 74 90]]