파이썬 torch.gather

t = torch.tensor([[1, 2, 3], [4, 5, 6]])

위를 출력하면

위에 처럼 나온다. 

t = torch.tensor([[1, 2, 3], [4, 5, 6]]).gather(0, torch.tensor([[0, 1]]))

위를 출력하면

위에처럼 1, 5가 나온다.

위와 같이 0차원에

차례대로 0번이랑 1번을 가져온것이다.

t = torch.tensor([[1, 2, 3], [4, 5, 6]]).gather(1, torch.tensor([[0, 1]]))

그럼 0차원을 1로 바꾸면 어떻게 될까

위에 처럼

1과 2가 나오게 된다. 그렇다면 4, 5, 6은 출력하지 못하는 걸까?

t = torch.tensor([[1, 2, 3], [4, 5, 6]]).gather(1, torch.tensor([[0, 1], [0, 1]]))

기존에는 0,1만 있었지만 추가로 0,1을 만들어줬다. 이 추가로 만들어 준것들이 

다음 텐서에 접근해준다.