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

8 lines
217 B
ReStructuredText

mindspore.ops.TopK
===================
.. py:class:: mindspore.ops.TopK(sorted=True)
沿最后一个维度查找 `k` 个最大元素和对应的索引。
更多参考详见 :func:`mindspore.ops.top_k`