forked from mindspore-Ecosystem/mindspore
27 lines
943 B
ReStructuredText
27 lines
943 B
ReStructuredText
|
mindspore.ops.InTopK
|
|||
|
|
====================
|
|||
|
|
|
|||
|
|
.. py:class:: mindspore.ops.InTopK(k)
|
|||
|
|
|
|||
|
|
判断目标标签是否在前 `k` 个预测中。
|
|||
|
|
|
|||
|
|
**参数：**
|
|||
|
|
|
|||
|
|
- **k** (int) - 指定 `k` 的值。
|
|||
|
|
|
|||
|
|
**输入：**
|
|||
|
|
|
|||
|
|
- **x1** (Tensor) - 2维Tensor，对样本的预测。数据类型支持float16或float32。
|
|||
|
|
- **x2** (Tensor) - 1维Tensor，样本的标签。数据类型为int32。 `x2` 的大小必须与 `x1` 第一维度的大小相同。 `x2` 取值不可为负且必须小于或等于 `x1` 第二维度的大小。
|
|||
|
|
|
|||
|
|
**输出：**
|
|||
|
|
|
|||
|
|
1维的bool类型Tensor，与 `x2` shape相同。对于样本 `i` 在 `x2` 中的标签, 如果其在 `x1` 对样本 `i` 的前 `k` 个预测中，则输出值为True，否则为False。
|
|||
|
|
|
|||
|
|
**异常：**
|
|||
|
|
|
|||
|
|
- **TypeError** - `k` 不是int类型。
|
|||
|
|
- **TypeError** - `x1` 或 `x2` 不是Tensor。
|
|||
|
|
- **TypeError** - `x1` 的数据类型非float16或float32。
|
|||
|
|
|