torch.gather 설명

원본은 킹텍오버플로우(클릭)

 

 

 

그림으로 보면 쉽게 이해가 된다. dim index가 0이 아닌 1부터 시작되는 것만 주의!