tensorflow的tf.reduce

tf.reduce_mean函数的作用是求平均值。第一个参数是一个集合，可以是列表、二维数组和多维数组。第二个参数指定在哪个维度上面求平均值。默认对所有的元素求平均。tf.reduce_mean
比如，下面是对所有元素求平均值：

x = tf.constant(
[[1., 1.],
[2., 2.]])tf.reduce_mean(x)  # 1.5

当指定第二个参数的时候，指定axis=0，表示沿着‘跨行’的方向求平均。对axis的理解可以参考一下这篇博文

m1 = tf.reduce_mean(x, axis=0)

结果为：[1.5, 1.5]

第二个参数不仅可以是一个数，也可以是一个数字，里面的数字表示指定所有的的轴的方向。
比如

xx = tf.constant([[[1., 1, 1],[2., 2, 2]],[[3, 3, 3],[4, 4, 4]]])
m3 = tf.reduce_mean(xx, [0, 1]) # [2.5 2.5 2.5]

上面是一个三维数组， xx的shape为（2，2，3），可以想象为三个2x2的二维数组叠加在一起形成一个2x2x3的立体，也就是三个面叠加。现在的axis为[0, 1]，表示对第1和第2轴的方向求平均值, 也就是分别对每一个面求平均。
第一个面为：
[[1., 2],
[3., 4]]
平均值为2.5

第二个面为
[[1., 2],
[3., 4]]
平均值为2.5

第三个面一样

完整的代码

import tensorflow as tfx = tf.constant([[1., 1.],[2., 2.]])
tf.reduce_mean(x)  # 1.5
m1 = tf.reduce_mean(x, axis=0)  # [1.5, 1.5]
m2 = tf.reduce_mean(x, 1)  # [1.,  2.]xx = tf.constant([[[1., 1, 1],[2., 2, 2]],[[3, 3, 3],[4, 4, 4]]])
m3 = tf.reduce_mean(xx, [0, 1]) # [2.5 2.5 2.5]with tf.Session() as sess:sess.run(tf.global_variables_initializer())print(sess.run(m1))print(sess.run(m2))print(xx.get_shape())print(sess.run(m3))

执行结果：
[1.5 1.5]
[1. 2.]
(2, 2, 3)
[2.5 2.5 2.5]

tensorflow的tf.reduce_mean函数相关推荐

TensorFlow中 tf.space_to_depth()函数的用法
目录一.函数定义二.解释范例三.代码验证一.函数定义通俗易懂些,就是把输入为[batch, height, width, channels]形式的Tensor,其在height和width维 ...
tensorflow中tf.get_variable()函数详解
如果变量存在,函数tf.get_variable()会返回现有的变量:如果变量不存在,会根据给定形状和初始值创建一个新的变量. def get_variable(name, shape=None, d ...
【tensorflow】tf.reshape函数说明：重塑张量
转载 [471]tf.reshape函数说明_周小董-CSDN博客 https://blog.csdn.net/xc_zhou/article/details/85342542 函数原型: tf.re ...
tensorflow 之 tf.tile()函数
这个函数是用来进行tensor维度扩展的,我们来看一下是怎么用的. 函数定义:tf.tile(input,multiples,name = None) input:表示输入tensor multipl ...
tf.graph函数
参考 [TensorFlow]tf.graph函数转载于:https://www.cnblogs.com/Aaron12/p/10036263.html
tf.Graph()函数
TensorFlow是谷歌基于DistBelief进行研发的第二代人工智能学习系统,其命名来源于本身的运行原理.Tensor(张量)意味着N维数组,Flow(流)意味着基于数据流图的计算,Tensor ...
tensorflow学习之常用函数总结：tensorflow官方例子中的诸如tf.reduce_mean()这类函数
前言 tensorflow官网给的例子用到了很多函数,然后并没有具体说明,还要自己去翻文档,有些函数是很常用的,下面来一一总结. 正文一,tensorflow中有一类在tensor的某一维度上求值的 ...
tensorflow reduce系列函数(tf.reduce_mean, tf.reduce_sum, tf.reduce_prod, tf.reduce_max, tf.reduce_min)
简而言之,reduce系列的函数都可在张量指定的维度上操作目录输入参数 tf.reduce_all 在boolean张量的维度上计算元素的 "逻辑和" tf.reduce_ ...
【TensorFlow】tf.nn.softmax_cross_entropy_with_logits 函数：求交叉熵损失
[TensorFlow]tf.nn.softmax_cross_entropy_with_logits的用法_xf__mao的博客-CSDN博客 https://blog.csdn.net/mao_x ...

tensorflow的tf.reduce_mean函数

tensorflow的tf.reduce_mean函数相关推荐

最新文章

热门文章