mindspore.numpy.mean

查看源文件
mindspore.numpy.mean(a, axis=None, keepdims=False, dtype=None)[源代码]

沿指定轴计算算术平均值。

返回数组元素的平均值。默认情况下，平均值是在展平的数组上计算的，否则在指定的轴上计算。

说明

不支持NumPy参数 out 。 在GPU上，支持的数据类型有np.float16和np.float32。

参数：

  • a (Tensor) - 包含要计算均值的数字的输入Tensor。如果 a 不是数组，将尝试进行转换。

  • axis (Union[int, tuple(int), None], 可选) - 计算均值的所沿的一个或多个轴。 默认计算展平数组的均值。 如果这是一个整数tuple，将在多个轴上计算均值。默认值: None 。

  • keepdims (bool, 可选) - 如果设置为 True，减少的轴在结果中保留为大小为1的维度。 若使用此选项，结果会广播到和输入Tensor同一个维度数。默认值: False 。

  • dtype (mindspore.dtype, 可选) - 默认值: None 。覆盖输出Tensor的dtype。

返回：

Tensor或标量，包含元素为所得均值的数组。

异常：

  • ValueError - 如果 axes 的范围超过 [-a.ndim, a.ndim) ，或如果 axes 包含重复项。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore.numpy as np
>>> a = np.arange(6, dtype='float32')
>>> output = np.mean(a, 0)
>>> print(output)
2.5