tf.reduce_mean

tf.reduce_mean(x) ==> 2.5 #如果不指定第二个参数,那么就在所有的元素中取平均值

tf.reduce_mean(x, 0) ==> [2., 3.] #指定第二个参数为0,则第一维的元素取平均值,即每一列求平均值

tf.reduce_mean(x, 1) ==> [1.5, 3.5] #

指定第二个参数为1,则第二维的元素取平均值,即每一行求平均值

   

   

   

posted @ 2019-01-18 10:53  simple_wxl  阅读(149)  评论(0编辑  收藏  举报