mindspore.mint.mean =================== .. image:: https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/master/resource/_static/logo_source.svg :target: https://gitee.com/mindspore/mindspore/blob/master/docs/api/api_python/mint/mindspore.mint.func_mean.rst :alt: 查看源文件 .. py:function:: mindspore.mint.mean(input, *, dtype=None) 移除输入所有维度,返回 `input` 中所有元素的平均值。 参数: - **input** (Tensor[Number]) - 输入Tensor,其数据类型为数值型。shape: :math:`(N, *)` ,其中 :math:`*` 表示任意数量的附加维度。 关键字参数: - **dtype** (:class:`mindspore.dtype`, 可选) - 期望输出Tensor的类型。默认值: ``None`` 。 返回: Tensor。 异常: - **TypeError** - `input` 不是Tensor。 .. py:function:: mindspore.mint.mean(input, dim, keepdim=False, *, dtype=None) :noindex: 默认情况下,移除输入所有维度,返回 `input` 中所有元素的平均值。也可仅缩小指定维度 `dim` 大小至1。 `keepdim` 控制输出和输入的维度是否相同。 .. note:: Tensor类型的 `dim` 仅用作兼容旧版本,不推荐使用。 参数: - **input** (Tensor[Number]) - 输入Tensor,其数据类型为数值型。shape: :math:`(N, *)` ,其中 :math:`*` 表示任意数量的附加维度。 - **dim** (Union[int, tuple(int), list(int), Tensor]) - 要减少的维度。只允许常量值。假设 `input` 的秩为r,取值范围[-r,r)。 - **keepdim** (bool) - 如果为 ``True`` ,则保留缩小的维度,大小为1。否则移除维度。默认值: ``False`` 。 关键字参数: - **dtype** (:class:`mindspore.dtype`, 可选) - 期望输出Tensor的类型。默认值: ``None`` 。 返回: Tensor。 - 如果 `dim` 为int,取值为1,并且 `keepdim` 为 ``False`` ,则输出的shape为 :math:`(input_0, input_2, ..., input_R)` 。 - 如果 `dim` 为tuple(int)或list(int),取值为(1, 2),并且 `keepdim` 为 ``False`` ,则输出Tensor的shape为 :math:`(input_0, input_3, ..., input_R)` 。 - 如果 `dim` 为一维Tensor,例如取值为[1, 2],并且 `keepdim` 为 ``False`` ,则输出Tensor的shape为 :math:`(input_0, input_3, ..., input_R)` 。 异常: - **TypeError** - `input` 不是Tensor。 - **TypeError** - `dim` 不是以下数据类型之一:int、tuple、list或Tensor。 - **TypeError** - `keepdim` 不是bool类型。 - **ValueError** - `dim` 超出范围。