脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → tensorflow tf.reduce_mean

tensorflow中tf.reduce_mean函数的使用

 更新时间：2020年04月19日 11:28:41   作者：-牧野-  
这篇文章主要介绍了tensorflow中tf.reduce_mean函数的使用，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧

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实现多语言朗读与单词选择测验

    在数字化教育日益普及的今天,开发一款能够支持多语言朗读和单词选择测验的程序,对于语言学习者来说无疑是一个巨大的福音,下面我们就来用Python实现一个这样的小程序吧
    2025-03-03
  • Django实现将一个字典传到前端显示出来

    Django实现将一个字典传到前端显示出来

    这篇文章主要介绍了Django实现将一个字典传到前端显示出来，具有很好的参考价值，希望
    2020-04-04
  • Python可视化Matplotlib介绍和简单图形的绘制

    Python可视化Matplotlib介绍和简单图形的绘制

    这篇文章主要介绍了Python可视化Matplotlib介绍和简单图形的绘制，文中附含详细示例代码，有需要的朋友可以借鉴参考下，希望能够有所帮助
    2021-09-09
  • python 使用pandas计算累积求和的方法

    python 使用pandas计算累积求和的方法

    今天小编就为大家分享一篇python 使用pandas计算累积求和的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2019-02-02
  • 5行Python代码实现图像分割的步骤详解

    5行Python代码实现图像分割的步骤详解

    这篇文章主要介绍了5行Python代码实现图像分割的步骤详解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2020-05-05
  • Python如何用字典完成匹配任务

    Python如何用字典完成匹配任务

    在生物信息学领域,经常需要根据基因名称匹配其对应的编号,本文介绍了一种通过字典进行基因名称与编号匹配的方法,首先定义一个空列表存储对应编号,对于字典中不存在的基因名称,其编号默认为0
    2024-09-09
  • python快速查找算法应用实例

    python快速查找算法应用实例

    这篇文章主要介绍了python快速查找算法,是一个比较经典的算法应用,有一定的借鉴价值,需要的朋友可以参考下
    2014-09-09
  • 使用Python随机生成数据的方法

    使用Python随机生成数据的方法

    这篇文章主要介绍了使用Python随机生成数据的方法,在日常开发中竟然会遇到需要测试大量数据的地方,那么随机生成数据就可以有效的加快我们的效率,通过Python_Faker生成测试数据需要安装Faker包,需要的朋友可以参考下
    2023-10-10
  • python中的json数据和pyecharts模块入门示例教程

    python中的json数据和pyecharts模块入门示例教程

    JSON是一种轻量级的数据交互格式。可以按照.JSON指定的格式去组织和封装数据，这篇文章主要介绍了python中的json数据和pyecharts模块入门,需要的朋友可以参考下
    2022-12-12
  • PYQT5设置textEdit自动滚屏的方法

    PYQT5设置textEdit自动滚屏的方法

    今天小编就为大家分享一篇PYQT5设置textEdit自动滚屏的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2019-06-06

最新评论