2.3K Star 8.1K Fork 4.3K

GVPMindSpore / mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.ops.func_var_mean.rst 2.16 KB
一键复制 编辑 原始数据 按行查看 历史

mindspore.ops.var_mean

.. py:function:: mindspore.ops.var_mean(input, axis=None, ddof=0, keepdims=False)

    默认情况下,输出Tensor各维度上的方差和均值,也可以对指定维度求方差和均值。如果 `axis` 是维度列表,则计算对应维度的方差和均值。

    .. note::
        如果 `ddof` 是0、1、True或False,支持的平台只有 `Ascend` 和 `CPU` 。其他情况下,支持平台是 `Ascend` 、 `GPU` 和 `CPU` 。

    参数:
        - **input** (Tensor[Number]) - 输入Tensor,其数据类型为数值型。shape: :math:`(N, *)` ,其中 :math:`*` 表示任意数量的附加维度。
        - **axis** (Union[int, tuple(int)],可选) - 要进行归约计算的维度。只允许常量值。假设 `input` 的秩为r,取值范围[-r,r)。默认值: ``None`` ,计算所有元素的均值和方差。
        - **ddof** (Union[int, bool],可选) - δ自由度。如果为整数,计算中使用的除数是 :math:`N - ddof` ,其中 :math:`N` 表示元素的数量。如果为 ``True`` ,使用Bessel校正。如果是 ``False`` ,使用偏置估计来计算方差。默认值: ``0`` 。
        - **keepdims** (bool,可选) - 是否保留输出Tensor的维度。如果为 ``True`` ,则保留缩小的维度,大小为1。否则移除维度。默认值: ``False`` 。

    返回:
        包含方差和均值的tuple。
        假设输入 `input` 的shape为 :math:`(x_0, x_1, ..., x_R)` :

        - 如果 `axis` 为(),且 `keepdims` 为 ``False`` ,则输出一个零维Tensor,表示输入Tensor `input` 中所有元素的方差。
        - 如果 `axis` 为int,取值为1,并且 `keepdims` 为 ``False`` ,则输出的shape为 :math:`(x_0, x_2, ..., x_R)` 。
        - 如果 `axis` 为tuple(int)或list(int),取值为(1, 2),并且 `keepdims` 为 ``False`` ,则输出Tensor的shape为 :math:`(x_0, x_2, ..., x_R)` 。

    异常:
        - **TypeError** - `input` 不是Tensor。
        - **TypeError** - `axis` 不是以下数据类型之一:None、int或tuple。
        - **TypeError** - `keepdims` 不是bool类型。
        - **ValueError** - `axis` 超出范围。
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
master

搜索帮助