如何在 pytorch 中實現 tf.nn.in_top_k

Question

我想在tf.nn.in_top_k中實現 tf.nn.in_top_k。 這是tf.nn.in_top_k的鏈接，

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

它將 k 處的精度計算為 bool 張量，並將返回 bool 類型的張量。

tf.nn.in_top_k

不知道pytorch里有沒有類似的api？

Answer 1

AFAIK in_top_k沒有內置等效的in_top_k函數。 寫一個相對簡單。 例如

def in_top_k(targets, preds, k):
    topk = preds.topk(k)[1]
    return (targets.unsqueeze(1) == topk).any(dim=1)