Tensor 내 특정 값 찾기

Pytorch tensor 내 특정 값과 해당되는 index가 필요할 경우 아래 코드로 확인할 수 있다.

 

x = torch.Tensor([11, 22, 33, 22])
print((x==22).nonzero(as_tuple=True)[0])

 

https://stackoverflow.com/questions/47863001/how-pytorch-tensor-get-the-index-of-specific-value

How Pytorch Tensor get the index of specific value

 

 

https://normal-engineer.tistory.com/301

[PyTorch] 특정 조건에 맞는 텐서 출력/인덱싱 등