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 相关文章推荐
用matplotlib画等高线图详解
Dec 14 Python
Python 16进制与中文相互转换的实现方法
Jul 09 Python
Python设计模式之原型模式实例详解
Jan 18 Python
python写日志文件操作类与应用示例
Jul 01 Python
selenium+PhantomJS爬取豆瓣读书
Aug 26 Python
Django中提示消息messages的设置方式
Nov 15 Python
python GUI编程(Tkinter) 创建子窗口及在窗口上用图片绘图实例
Mar 04 Python
python shapely.geometry.polygon任意两个四边形的IOU计算实例
Apr 12 Python
简单了解python列表和元组的区别
May 14 Python
用Python开发app后端有优势吗
Jun 29 Python
Python基础之教你怎么在M1系统上使用pandas
May 08 Python
Pytorch数据读取之Dataset和DataLoader知识总结
May 23 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 模拟登陆MSN并获得用户信息
2009/05/16 PHP
Zend Framework基本页面布局分析
2016/03/19 PHP
smarty的section嵌套循环用法示例
2016/05/28 PHP
PHP中spl_autoload_register()函数用法实例详解
2016/07/18 PHP
PHP从二维数组得到N层分类树的实现代码
2016/10/11 PHP
Yii2实现多域名跨域同步登录退出
2017/02/04 PHP
PHP图片裁剪与缩放示例(无损裁剪图片)
2017/02/08 PHP
Laravel 关联模型-关联新增和关联更新的方法
2019/10/10 PHP
js函数使用技巧之 setTimeout(function(){},0)
2009/02/09 Javascript
使用Node.js实现一个简单的FastCGI服务器实例
2014/06/09 Javascript
jQuery中:only-child选择器用法实例
2015/01/03 Javascript
JavaScript操作cookie类实例
2015/03/31 Javascript
JavaScript获取网页支持表单字符集的方法
2015/04/02 Javascript
使用AngularJS实现可伸缩的页面切换的方法
2015/06/19 Javascript
jQuery实现向下滑出的平滑下拉菜单效果
2015/08/21 Javascript
JavaScript每天必学之数组和对象部分
2016/09/17 Javascript
Javascript DOM事件操作小结(监听鼠标点击、释放,悬停、离开等)
2017/01/20 Javascript
微信小程序 WebSocket详解及应用
2017/01/21 Javascript
Bootstrap模态框使用详解
2017/02/15 Javascript
Bootstrap导航中表单简单实现代码
2017/03/06 Javascript
深入浅析Nodejs的Http模块
2017/06/20 NodeJs
微信小程序使用slider设置数据值及switch开关组件功能【附源码下载】
2017/12/09 Javascript
JS实现电商商品展示放大镜特效
2020/01/07 Javascript
python控制台显示时钟的示例
2014/02/24 Python
Python中index()和seek()的用法(详解)
2017/04/27 Python
Python使用pip安装pySerial串口通讯模块
2018/04/20 Python
Python学习_几种存取xls/xlsx文件的方法总结
2018/05/03 Python
django 在原有表格添加或删除字段的实例
2018/05/27 Python
浅谈pandas筛选出表中满足另一个表所有条件的数据方法
2019/02/08 Python
使用Python的networkx绘制精美网络图教程
2019/11/21 Python
python实现对列表中的元素进行倒序打印
2019/11/23 Python
手工制作的意大利礼服鞋:Ace Marks
2018/12/15 全球购物
护理职业应聘自荐书
2013/09/29 职场文书
小学数学课后反思
2014/04/23 职场文书
团员年度个人总结
2015/02/26 职场文书
实习单位意见
2015/06/04 职场文书