tf.reduce_sum

用于在某个维度求和

reduce_sum(
    input_tensor,
    axis=None,
    keep_dims=False,
    name=None,
    reduction_indices=None
)

input_tensor:表示输入 
axis:表示在那个维度进行sum操作。 
keep_dims:表示是否保留原始数据的维度,False相当于执行完后原始数据就会少一个维度。 
reduction_indices:为了跟旧版本的兼容,现在已经不使用了。 

例子:

import tensorflow as tf
import numpy as np
a = [[[1,2], [3,4], [5,6],[7,8]]]

with tf.Session() as session:
    print("========在所有维度求和===================")
    print(session.run(tf.reduce_sum(a)))
    print("========在0维求和========================")
    print(session.run(tf.reduce_sum(a, axis=0)))
    print("========在0维求和,不降维================")
    print(session.run(tf.reduce_sum(a, axis=0, keepdims=True)))
    print("========在1维求和========================")
    print(session.run(tf.reduce_sum(a, axis=1)))
    print("========在1维求和,不降维================")
    print(session.run(tf.reduce_sum(a, axis=1, keepdims=True)))
    print("========在2维求和========================")
    print(session.run(tf.reduce_sum(a, axis=2)))
    print("========在2维求和,不降维================")
    print(session.run(tf.reduce_sum(a, axis=2, keepdims=True)))

输出:

========在所有维度求和===================
36
========在0维求和========================
[[1 2]
 [3 4]
 [5 6]
 [7 8]]
========在0维求和,不降维================
[[[1 2]
  [3 4]
  [5 6]
  [7 8]]]
========在1维求和========================
[[16 20]]
========在1维求和,不降维================
[[[16 20]]]
========在2维求和========================
[[ 3  7 11 15]]
========在2维求和,不降维================
[[[ 3]
  [ 7]
  [11]
  [15]]] 

结合例子和输出我们说明下。a是一个三维向量。

在第0维,只有一个元素,这个元素是一个二维向量

[[1 2]
 [3 4]
 [5 6]
 [7 8]] 

所以在第0维求和,就是这个元素本身。


在第1维,有四个元素,每个元素是一维向量,分别是

[1,2], [3,4], [5,6],[7,8]

所以在第1维求和,对应位置相加,就是

[[16 20]]
所谓降维与不降维可以简单的理解为去不去掉一个括号(维度)



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值