tf.reduce_mean的用法

最新推荐文章于 2022-04-08 12:15:11 发布

UESTC_C2_403

最新推荐文章于 2022-04-08 12:15:11 发布

阅读量1.3w

点赞数 2

分类专栏： tensorflow用法

本文链接：https://blog.csdn.net/UESTC_C2_403/article/details/72190196

版权

tensorflow用法专栏收录该内容

64 篇文章 9 订阅

订阅专栏

import tensorflow as tf;
import numpy as np;

A = np.array([[1,2], [3,4]])

with tf.Session() as sess:
	print sess.run(tf.reduce_mean(A))
	print sess.run(tf.reduce_mean(A, axis=0))
	print sess.run(tf.reduce_mean(A, axis=1))

输出：
2 #整体的平均值
[2 3] #按列求得平均
[1 3] #按照行求得平均
如果没有axis求得就是整体的平均，