mindspore/docs/api/api_python/ops/mindspore.ops.Argmax.rst

9 lines
231 B
ReStructuredText

mindspore.ops.Argmax
=====================
.. py:class:: mindspore.ops.Argmax(axis=-1, output_type=mstype.int32)
返回输入Tensor在指定轴上的最大值索引。
详情请查看 :func:`mindspore.ops.argmax`