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 相关文章推荐
Python3.2中Print函数用法实例详解
May 19 Python
Python实现递归遍历文件夹并删除文件
Apr 18 Python
10 行 Python 代码教你自动发送短信(不想回复工作邮件妙招)
Oct 11 Python
5分钟 Pipenv 上手指南
Dec 20 Python
浅谈python3.6的tkinter运行问题
Feb 22 Python
python字符串循环左移
Mar 08 Python
pyqt弹出新对话框,以及关闭对话框获取数据的实例
Jun 18 Python
python jenkins 打包构建代码的示例代码
Nov 29 Python
Python的对象传递与Copy函数使用详解
Dec 26 Python
使用wxpy实现自动发送微信消息功能
Feb 28 Python
Python自动化测试中yaml文件读取操作
Aug 20 Python
Python安装使用Scrapy框架
Apr 12 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循环检测目录是否存在并创建(循环创建目录)
2011/01/06 PHP
PHP单例模式定义与使用实例详解
2017/02/06 PHP
phpstudy默认不支持64位php的解决方法
2017/02/20 PHP
PHP7 标准库修改
2021/03/09 PHP
JavaScript脚本性能的优化方法
2007/02/02 Javascript
setInterval,setTimeout与jquery混用的问题
2013/04/08 Javascript
jQuery实现鼠标移到元素上动态提示消息框效果
2013/10/20 Javascript
jquery 插件实现多行文本框[textarea]自动高度
2015/03/04 Javascript
基于angularjs实现图片放大镜效果
2016/08/31 Javascript
Vue.js组件tab实现选项卡切换
2020/03/23 Javascript
es6中的解构赋值、扩展运算符和rest参数使用详解
2017/09/28 Javascript
vue实现选项卡及选项卡切换效果
2018/04/24 Javascript
微信小程序云开发如何使用npm安装依赖
2019/05/18 Javascript
Vue-cli3多页面配置详解
2020/03/22 Javascript
[32:30]夜魇凡尔赛茶话会 第一期01:谁是卧底
2021/03/11 DOTA
Python 含参构造函数实例详解
2017/05/25 Python
Python3 处理JSON的实例详解
2017/10/29 Python
python实现对指定输入的字符串逆序输出的6种方法
2018/04/26 Python
python print输出延时,让其立刻输出的方法
2019/01/07 Python
ansible动态Inventory主机清单配置遇到的坑
2020/01/19 Python
Python pyautogui模块实现鼠标键盘自动化方法详解
2020/02/17 Python
python str字符串转uuid实例
2020/03/03 Python
python suds访问webservice服务实现
2020/06/26 Python
Python实现EM算法实例代码
2020/10/04 Python
纯CSS3实现绘制各种图形实现代码详细整理
2012/12/26 HTML / CSS
使用CSS3制作响应式导航菜单的方法
2015/07/12 HTML / CSS
罗技英国官方网站:Logitech UK
2020/11/03 全球购物
学院领导推荐信
2013/10/30 职场文书
项目经理岗位职责
2013/11/11 职场文书
咖啡厅创业计划书范本
2014/01/22 职场文书
安全承诺书范文
2014/03/26 职场文书
节电标语大全
2014/06/23 职场文书
世界读书日的活动方案
2014/08/20 职场文书
二年级语文下册复习计划
2015/01/19 职场文书
人民币使用说明书
2019/04/17 职场文书
经典励志格言:每日一句,让你每天充满能量
2019/08/16 职场文书