Pytorch張量如何得到具體值的索引

Question

在python列表中，我們可以使用list.index(somevalue)。 pytorch如何做到這一點？
例如：

a=[1,2,3] 
    print(a.index(2)) 

然後，1將輸出。 pytorch tensor如何在不將其轉換爲python列表的情況下執行此操作？

Answer 1

我認爲沒有從list.index()到pytorch函數的直接翻譯。但是，您可以使用tensor==number，然後使用nonzero()函數獲得類似的結果。例如：

t = torch.Tensor([1, 2, 3]) 
print ((t == 2).nonzero()) 

這段代碼返回

[大小1×1的torch.LongTensor]