mindspore.ops.TopK
- class mindspore.ops.TopK(sorted=True)[source]
Finds values and indices of the k largest entries along the last dimension.
Refer to
mindspore.ops.top_k()
for more detail.- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> 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]))