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]