[Tensor] Tensor value index / Tensor에서 특정 값에 대응되는 index 번호 반환하기

(tensor == target_value).nonzero(as_tuple=True)

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

How Pytorch Tensor get the index of specific value