码迷,mamicode.com
首页 > 其他好文 > 详细

tf.reduce_mean

时间:2018-06-07 19:42:11      阅读:118      评论:0      收藏:0      [点我收藏+]

标签:参数   const   3.5   指定   TE   type   class   highlight   oat   

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

标签:参数   const   3.5   指定   TE   type   class   highlight   oat   

原文地址:https://www.cnblogs.com/zenan/p/9151930.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!