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 中文乱码问题深入分析
Mar 13 Python
Python三级目录展示的实现方法
Sep 28 Python
Python基于Socket实现的简单聊天程序示例
Aug 05 Python
python中将函数赋值给变量时需要注意的一些问题
Aug 18 Python
定制FileField中的上传文件名称实例
Aug 23 Python
Python实现购物车程序
Apr 16 Python
pandas表连接 索引上的合并方法
Jun 08 Python
Python3 使用cookiejar管理cookie的方法
Dec 28 Python
Python math库 ln(x)运算的实现及原理
Jul 17 Python
python实现京东订单推送到测试环境,提供便利操作示例
Aug 09 Python
Python爬虫爬取煎蛋网图片代码实例
Dec 16 Python
Python采集股票数据并制作可视化柱状图
Apr 04 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中static关键字原理的学习研究分析
2011/07/18 PHP
ThinkPHP模板判断输出Present标签用法详解
2014/06/30 PHP
php绘制圆形的方法
2015/01/24 PHP
PHP 正则表达式小结
2015/02/12 PHP
PHP设计模式之模板模式定义与用法详解
2018/12/20 PHP
Table冻结表头示例代码
2013/08/20 Javascript
JS弹出可拖拽可关闭的div层完整实例
2015/02/13 Javascript
cookie的secure属性详解
2015/04/08 Javascript
text-align:justify实现文本两端对齐 兼容IE
2015/08/19 Javascript
JQuery validate插件Remote用法大全
2016/05/15 Javascript
Javascript中prototype的使用详解
2016/06/18 Javascript
基于js里调用函数时,函数名带括号和不带括号的区别
2016/07/28 Javascript
jQuery源码解读之extend()与工具方法、实例方法详解
2017/03/30 jQuery
浅谈JsonObject中的key-value数据解析排序问题
2017/12/06 Javascript
JavaScript中常见内置函数用法示例
2018/05/14 Javascript
nodejs高大上的部署方式(PM2)
2018/09/11 NodeJs
Javascript地址引用代码实例解析
2020/02/25 Javascript
Python中max函数用法实例分析
2015/07/17 Python
Collatz 序列、逗号代码、字符图网格实例
2017/06/22 Python
Flask模板引擎之Jinja2语法介绍
2019/06/26 Python
Python 离线工作环境搭建的方法步骤
2019/07/29 Python
python 使用pdfminer3k 读取PDF文档的例子
2019/08/27 Python
Python实现鼠标自动在屏幕上随机移动功能
2020/03/14 Python
在Pytorch中使用Mask R-CNN进行实例分割操作
2020/06/24 Python
python中strip(),lstrip(),rstrip()函数的使用讲解
2020/11/17 Python
Herve Leger官网:标志性绷带连衣裙等
2018/12/26 全球购物
大学学习生活感言
2014/01/18 职场文书
家庭教育先进个人事迹材料
2014/01/24 职场文书
竞选班干部演讲稿
2014/04/24 职场文书
公司董事长助理工作职责
2014/07/12 职场文书
光学与应用专业毕业生求职信
2014/09/01 职场文书
入党积极分子考察意见
2015/06/02 职场文书
电影雷锋观后感
2015/06/10 职场文书
呼啸山庄读书笔记
2015/06/29 职场文书
导游词之南京夫子庙
2019/12/09 职场文书
python开发的自动化运维工具ansible详解
2021/08/07 Python