2.6K Star 8.6K Fork 4.8K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.mint.any.rst 1.81 KB
一键复制 编辑 原始数据 按行查看 历史
张栩浩 提交于 2024-06-29 15:03 . mint.any文档修复

mindspore.mint.any

.. py:function:: mindspore.mint.any(input, dim=None, keepdim=False)

    默认情况下,通过对维度中所有元素进行“逻辑或”来减少 `input` 的维度。也可以沿 `dim` 减少 `input` 的维度。通过控制 `keepdim` 来确定输出和输入的维度是否相同。

    .. note::
        Tensor类型的 `dim` 仅用作兼容旧版本,不推荐使用。

    参数:
        - **input** (Tensor) - 输入Tensor,shape是 :math:`(N, *)` ,其中 :math:`*` 表示任意数量的附加维度。
        - **dim** (Union[int, tuple(int), list(int), Tensor], 可选) - 要减少的维度。只允许常量值。假设 `input` 的秩为r,取值范围[-r,r)。默认值: ``None`` ,缩小所有维度。
        - **keepdim** (bool, 可选) - 如果为 ``True`` ,则保留缩小的维度,大小为1。否则移除维度。默认值: ``False`` 。

    返回:
        Tensor,数据类型是bool。

        - 如果 `dim` 为 ``None`` ,且 `keepdim` 为 ``False`` ,则输出一个零维Tensor,表示输入Tensor中所有元素进行“逻辑或”。
        - 如果 `dim` 为int,例如取值为2,并且 `keepdim` 为 ``False`` ,则输出的shape为 :math:`(input_1, input_3, ..., input_R)` 。
        - 如果 `dim` 为tuple(int)或list(int),例如取值为(2, 3),并且 `keepdim` 为 ``False`` ,则输出Tensor的shape为 :math:`(input_1, input_4, ..., input_R)` 。
        - 如果 `dim` 为一维Tensor,例如取值为[2, 3],并且 `keepdim` 为 ``False`` ,则输出Tensor的shape为 :math:`(input_1, input_4, ..., input_R)` 。

    异常:
        - **TypeError** - `keepdim` 不是bool类型。
        - **TypeError** - `input` 不是Tensor。
        - **TypeError** - `dim` 不是以下数据类型之一:int、tuple、list或Tensor。
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
r2.3.1

搜索帮助