mindspore.ops.InTopK

class mindspore.ops.InTopK(k)[源代码]

判断目标标签是否在前 k 个预测中。

更多参考详见 mindspore.ops.intopk()

支持平台:

Ascend GPU CPU

样例:

>>> x1 = Tensor(np.array([[1, 8, 5, 2, 7], [4, 9, 1, 3, 5]]), mindspore.float32)
>>> x2 = Tensor(np.array([1, 3]), mindspore.int32)
>>> in_top_k = ops.InTopK(3)
>>> output = in_top_k(x1, x2)
>>> print(output)
[ True  False]