mindspore.ops.TopK
- class mindspore.ops.TopK(sorted=True)[源代码]
沿最后一个维度查找 k 个最大元素和对应的索引。
更多参考详见
mindspore.ops.top_k()
。- 支持平台:
Ascend
GPU
CPU
样例:
>>> from mindspore import Tensor >>> from mindspore import ops >>> import mindspore >>> input_x = Tensor([1, 2, 3, 4, 5], mindspore.float16) >>> k = 3 >>> values, indices = ops.TopK(sorted=True)(input_x, k) >>> print((values, indices)) (Tensor(shape=[3], dtype=Float16, value= [ 5.0000e+00, 4.0000e+00, 3.0000e+00]), Tensor(shape=[3], dtype=Int32, value= [4, 3, 2]))