遍歷 pytorch 中的張量

Question

我有兩個一維張量。 一個是預測向量，第二個是標簽向量。 我正在嘗試編寫一個循環來檢查向量之間的元素差異。 如果發現這樣的差異，我想做另一個操作，為簡單起見，假設我想打印（“發現差異”）。 到目前為止，我想出了這個但我得到了一個錯誤：預期的標量類型字節的 object 但參數＃2“其他”的標量類型浮點數。 我會很感激這里的幫助。 也許有一些更有效的方法可以做到這一點，沒有循環。

for i in enumerate(t1):
    if t1[i] != t2[i]:
        print("Diff spotted")

Answer 1

您可以使用 pytorch 中的eq() function 來檢查張量是否在元素方面相同。 對於與標簽元素相同的元素的每個索引，您將獲得True ：

for label in predictions.round().eq(labels):
    for element in label:
        if element == False:
            print("Diff spotted!")