23 lines
1.3 KiB
ReStructuredText
23 lines
1.3 KiB
ReStructuredText
mindspore.numpy.mean
|
||
====================
|
||
|
||
.. py:function:: mindspore.numpy.mean(a, axis=None, keepdims=False, dtype=None)
|
||
|
||
沿指定轴计算算术平均值。
|
||
|
||
返回数组元素的平均值。默认情况下，平均值是在展平的数组上计算的，否则在指定的轴上计算。
|
||
|
||
.. note::
|
||
不支持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` 包含重复项。 |