Pytorch Tensorが特定の値のインデックスを取得する方法

Question

pythonリストでは、list.index(somevalue)を使用できます。どのようにしてこれを行うことができますか？例えば
：

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

そして、1が出力されます。どのようにpytorchテンソルは、これをPythonリストに変換せずに行うことができますか？

Answer 1

list.index()からpytorch関数への直接的な翻訳はないと思います。ただし、tensor==number、次にnonzero()関数を使用すると、同様の結果が得られます。例えば：

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

コードのこの部分は

[サイズ1×1のtorch.LongTensor]

を返します