tf.reduce_sum函数中reduction_indices详解

TensorFlow 的python使用方式:
在tensorflow的使用中,经常会使用tf.reduce_mean,tf.reduce_sum等函数,函数中reduction_indices参数表示函数的处理维度:
reduction_indices参数的值默认的时候为None,默认把所有的数据求和,即结果是一维的。
reduction_indices参数的值为0的时候,是第0维对应位置相加。
reduction_indices参数的值为1的时候,是第1维对应位置相加。

图例:

è¿éåå¾çæè¿°

代码示例:

# tf.reduce_sum函数中reduction_indices参数的使用:
import numpy as np
import tensorflow as tf
a = np.array([[1, 2, 3], [4, 5, 6]])
b = tf.reduce_sum(a, reduction_indices=[0])
c = tf.reduce_sum(a, reduction_indices=[1])
d = tf.reduce_sum(a)

with tf.Session() as sess:
    # tf.reduce_sum函数中reduction_indices参数的使用:
    # reduction_indices表示函数的处理维度
    b_result, c_result, d_result = sess.run([b, c, d])
    print("原数据a:")
    print(a)
    print("a的形状:", a.shape)
    print("reduction_indices参数的值为0的时候:")
    print("b=tf.reduce_sum(a, reduction_indices=[0])=", b_result)
    print("reduction_indices参数的值为1的时候:")
    print("c=tf.reduce_sum(a, reduction_indices=[1])=", c_result)
    print("reduction_indices参数的值默认的时候为None,默认所有求和,即一维的:")
    print("d=tf.reduce_sum(a)=", d_result)

运行结果如下:

å¨è¿éæå¥å¾çæè¿°

参考:https://blog.csdn.net/qq_33096883/article/details/77479766

https://blog.csdn.net/u014772246/article/details/84973358

  • 1
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值