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]: <tf.Tensor: shape=(2,), dtype=float32, numpy=array([1., 2.], dtype=float32)>
发布了70 篇原创文章 · 获赞 46 · 访问量 6907
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 技术黑板 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览