tensorflow2.1中求平均值函数tf.reduce_mean()的用法
tf.reduce_mean 函数用于计算张量tensor沿着指定的数轴（tensor的某一维度）上的的平均值，主要用作降维或者计算tensor（图像）的平均值。
reduce_mean
(input_tensor,axis=None,keepdims=False,name=None,
)
参数说明：
- input_tensor - 输入的待降维的tensor;
- axis - 指定的轴，如果不指定，则计算所有元素的均值;
- keepdims - 是否降维度，设置为True，输出的结果保持输入tensor的形状，设置为False，输出结果会降低维度;
- name - 操作的名称;
示例如下：
import tensorflow as tf
x = tf.constant([[1., 1.], [2., 2.]])
tf.reduce_mean(x) # 1.5
tf.reduce_mean(x, 0) # [1.5, 1.5]
tf.reduce_mean(x, 1)
# Out[29]: 本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！