mindspore.Tensor.argmax ======================= .. 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/mindspore/Tensor/mindspore.Tensor.argmax.rst :alt: 查看源文件 .. py:method:: mindspore.Tensor.argmax(axis=None, keepdims=False) -> Tensor 返回tensor在指定轴上的最大值索引。 参数: - **axis** (Union[int, None],可选) - 指定计算轴。如果为 ``None`` ,计算tensor中的所有元素。默认 ``None`` 。 - **keepdims** (bool,可选) - 输出tensor是否保留维度。默认 ``False`` 。 返回: Tensor .. py:method:: mindspore.Tensor.argmax(dim=None, keepdim=False) -> Tensor :noindex: 返回tensor在指定轴上的最大值索引。 参数: - **dim** (Union[int, None],可选) - 指定计算维度。如果为 ``None`` ,计算tensor中的所有元素。默认 ``None`` 。 - **keepdim** (bool,可选) - 输出tensor是否保留维度。默认 ``False`` 。 返回: Tensor