tf.reduce_mean
Posted zenan
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了tf.reduce_mean相关的知识，希望对你有一定的参考价值。
x = tf.constant([[1,2]
[3,4]],dtype=tf.float32)
tf.reduce_mean(x) ==> 2.5 #如果不指定第二个参数，那么就在所有的元素中取平均值
tf.reduce_mean(x, 0) ==> [2., 3.] #指定第二个参数为0，则第一维的元素取平均值，即每一列求平均值
tf.reduce_mean(x, 1) ==> [1.5, 3.5] #指定第二个参数为1，则第二维的元素取平均值，即每一行求平均值
以上是关于tf.reduce_mean的主要内容，如果未能解决你的问题，请参考以下文章