2.6K Star 8.6K Fork 4.8K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.mint.func_argmin.rst 807 Bytes
一键复制 编辑 原始数据 按行查看 历史
shenhaojing 提交于 2024-07-23 16:56 . aclnnArgmin

mindspore.mint.argmin

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

    返回输入Tensor在指定轴上的最小值索引。

    参数:
        - **input** (Tensor) - 输入Tensor。
        - **dim** (Union[int, None],可选) - 指定计算轴。如果是 ``None`` ,将会返回扁平化Tensor在指定轴上的最小值索引。默认值: ``None`` 。
        - **keepdim** (bool,可选) - 输出Tensor是否保留指定轴。如果 `dim` 是 ``None`` ,忽略该选项。默认值: ``False`` 。

    返回:
        Tensor,输出为指定轴上输入Tensor最小值的索引。

    异常:
        - **TypeError** - 如果 `keepdim` 的类型不是bool值。
        - **ValueError** - 如果 `dim` 的设定值超出了范围。
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
master

搜索帮助