Pytorch Tensor가 특정 값의 인덱스를 얻는 방법

Question

파이썬 목록에서 우리는 list.index(somevalue)을 사용할 수 있습니다. 파이크가 어떻게 이럴 수 있지? 예
:

이어서

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

, 1가 출력된다. 어떻게 파이톤 텐서가 파이썬리스트로 변환하지 않고 이것을 할 수 있습니까?

Answer 1

list.index()에서 pytorch 기능으로의 직접적인 번역은 없다고 생각합니다. 그러나 tensor==number 다음에 nonzero() 함수를 사용하면 비슷한 결과를 얻을 수 있습니다. 예를 들면 :는

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

이 코드 조각

1

[torch.LongTensor 크기가 1x1] 반환