Python 如何在pytorch中实现tf.nn.in_top_k

我想在pytorch中实现

tf.nn.in_top_k

。这里是

tf.nn.in\u top\u k

的链接

tf.math.in_top_k(
    targets, predictions, k, name=None
)

它以布尔张量的形式计算k处的精度，并返回布尔类型的张量

---

我想知道Pytork中是否有类似的api？

AFAIK在Pytork中没有内置的等效函数

。写一个比较简单。比如说

def在顶部（目标、PreD、k）：
topk=preds.topk（k）[1]
返回（targets.unsqueze（1）=topk.any（dim=1）