tensorflow中tf.reduce_mean函数的使用


Posted in Python onApril 19, 2020

tf.reduce_mean 函数用于计算张量tensor沿着指定的数轴(tensor的某一维度)上的的平均值,主要用作降维或者计算tensor(图像)的平均值。

reduce_mean(input_tensor,
        axis=None,
        keep_dims=False,
        name=None,
        reduction_indices=None)
  • 第一个参数input_tensor: 输入的待降维的tensor;
  • 第二个参数axis: 指定的轴,如果不指定,则计算所有元素的均值;
  • 第三个参数keep_dims:是否降维度,设置为True,输出的结果保持输入tensor的形状,设置为False,输出结果会降低维度;
  • 第四个参数name: 操作的名称;
  • 第五个参数 reduction_indices:在以前版本中用来指定轴,已弃用; 

以一个维度是2,形状是[2,3]的tensor举例:

import tensorflow as tf
 
x = [[1,2,3],
   [1,2,3]]
 
xx = tf.cast(x,tf.float32)
 
mean_all = tf.reduce_mean(xx, keep_dims=False)
mean_0 = tf.reduce_mean(xx, axis=0, keep_dims=False)
mean_1 = tf.reduce_mean(xx, axis=1, keep_dims=False)
 
 
with tf.Session() as sess:
  m_a,m_0,m_1 = sess.run([mean_all, mean_0, mean_1])
 
print m_a  # output: 2.0
print m_0  # output: [ 1. 2. 3.]
print m_1  #output: [ 2. 2.]

如果设置保持原来的张量的维度,keep_dims=True ,结果:

print m_a  # output: [[ 2.]]
print m_0  # output: [[ 1. 2. 3.]]
print m_1  #output: [[ 2.], [ 2.]]

类似函数还有:

  • tf.reduce_sum :计算tensor指定轴方向上的所有元素的累加和;
  • tf.reduce_max  :  计算tensor指定轴方向上的各个元素的最大值;
  • tf.reduce_all :  计算tensor指定轴方向上的各个元素的逻辑和(and运算);
  • tf.reduce_any:  计算tensor指定轴方向上的各个元素的逻辑或(or运算);

到此这篇关于tensorflow中tf.reduce_mean函数的使用的文章就介绍到这了,更多相关tensorflow tf.reduce_mean内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木! 

Python 相关文章推荐
深入探究Python中变量的拷贝和作用域问题
May 05 Python
举例讲解Django中数据模型访问外键值的方法
Jul 21 Python
python先序遍历二叉树问题
Nov 10 Python
Python科学画图代码分享
Nov 29 Python
Python中的defaultdict与__missing__()使用介绍
Feb 03 Python
python2与python3共存问题的解决方法
Sep 18 Python
django Admin文档生成器使用详解
Jul 22 Python
详解python中*号的用法
Oct 21 Python
解决Tensorflow占用GPU显存问题
Feb 03 Python
python函数定义和调用过程详解
Feb 09 Python
TensorFlow2.1.0最新版本安装详细教程
Apr 08 Python
不到20行实现Python代码即可制作精美证件照
Apr 24 Python
TensorFlow打印输出tensor的值
Apr 19 #Python
numpy库reshape用法详解
Apr 19 #Python
tensorflow常用函数API介绍
Apr 19 #Python
TensorFlow的reshape操作 tf.reshape的实现
Apr 19 #Python
pip安装tensorflow的坑的解决
Apr 19 #Python
查看已安装tensorflow版本的方法示例
Apr 19 #Python
在Anaconda3下使用清华镜像源安装TensorFlow(CPU版)
Apr 19 #Python
You might like
以实例全面讲解PHP中多进程编程的相关函数的使用
2015/08/18 PHP
PHP的简单跳转提示的实现详解
2019/03/14 PHP
php字符串函数 str类常见用法示例
2020/05/15 PHP
BOOM vs RR BO5 第一场 2.14
2021/03/10 DOTA
Maps Javascript
2007/01/22 Javascript
js汉字转拼音实现代码
2013/02/06 Javascript
JavaScript:Div层拖动效果实例代码
2013/08/06 Javascript
带左右箭头图片轮播的JS代码
2013/12/18 Javascript
js数组循环遍历数组内所有元素的方法
2014/01/18 Javascript
js获取当前日期时间及其它日期操作汇总
2016/03/08 Javascript
js删除局部变量的实现方法
2016/06/25 Javascript
Angular中使用ui router实现系统权限控制及开发遇到问题
2016/09/23 Javascript
针对后台列表table拖拽比较实用的jquery拖动排序
2016/10/10 Javascript
微信小程序 高德地图SDK详解及简单实例(源码下载)
2017/01/11 Javascript
JavaScript两个变量交换值的实现方法
2017/03/01 Javascript
JS实现合并json对象的方法
2017/10/10 Javascript
vue 实现 ios 原生picker 效果及实现思路解析
2017/12/06 Javascript
python中assert用法实例分析
2015/04/30 Python
Python使用xlrd模块操作Excel数据导入的方法
2015/05/26 Python
详解C++编程中一元运算符的重载
2016/01/19 Python
最近Python有点火? 给你7个学习它的理由!
2017/06/26 Python
Python3实现抓取javascript动态生成的html网页功能示例
2017/08/22 Python
python使用threading获取线程函数返回值的实现方法
2017/11/15 Python
python中WSGI是什么,Python应用WSGI详解
2017/11/24 Python
python实现多线程行情抓取工具的方法
2018/02/28 Python
numpy找出array中的最大值,最小值实例
2018/04/03 Python
python 获取页面表格数据存放到csv中的方法
2018/12/26 Python
Python Django 命名空间模式的实现
2019/08/09 Python
Python爬虫实现HTTP网络请求多种实现方式
2020/06/19 Python
Python3基于print打印带颜色字符串
2020/07/06 Python
我的长生果教学反思
2014/04/28 职场文书
经营目标管理责任书
2014/07/25 职场文书
员工拾金不昧表扬稿
2015/05/05 职场文书
干部考核工作总结
2015/08/12 职场文书
学习心得体会
2019/06/20 职场文书
Spring Boot 启动、停止、重启、状态脚本
2021/06/26 Java/Android