Tuple 에서 index 찾기

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

python list에서 특정 값의 index를 찾는 것 처럼 tensor에서 특정 값의 index 위치 정보를 얻고 싶으면 상단 command를 활용하면 된다. 아래는 이를 활용한 예제

new_sizes = tensor([4, 4, 0, 4, 4, 4, 3, 4, 4, 4, 0, 4, 4, 4, 0, 4, 4, 4, 0, 4, 4, 4, 3, 4,
        4, 4, 0, 4, 4, 4, 0, 4, 4, 4, 0, 4, 4, 4, 0, 4, 4, 5, 0, 4, 4, 4, 0, 4,
        4, 4, 0, 4, 4, 4, 0, 4, 4, 4, 3, 4, 4, 4, 3, 4, 4, 6, 3, 4, 4, 5, 3, 4,
        4, 4, 0, 4, 5, 5, 3, 5, 4, 4, 3, 4, 4, 4, 3, 4, 4, 5, 0, 4, 4, 4, 3, 4,
        4, 5, 3, 4, 4, 5, 0, 4, 4, 4, 3, 4, 4, 4, 3, 4, 4, 4, 3, 4, 4, 4, 3, 4])
        
print((new_sizes == 6).nonzero(as_tuple=True)) # 65