물건을 여러 개의 상자에 담고, 각 상자에는 번호가 부여되어 있는 상황을 상상해봅시다. 이때, 각 상자에 담긴 물건을 선택할 때 특정 상자의 번호를 참고하고 싶다고 가정해봅시다.
`torch.gather()` 함수는 이러한 상황에서 사용될 수 있습니다. 예를 들어, 우리가 상자에 담긴 물건의 목록과 각 상자의 번호가 주어진다면, `torch.gather()` 함수를 사용하여 특정 상자의 물건을 선택할 수 있습니다.
다음은 `torch.gather()` 함수의 예시입니다:
상자에 담긴 물건: [‘사과’, ‘바나나’, ‘딸기’, ‘오렌지’]
각 상자의 번호: [2, 0, 1, 3]
우리가 2번 상자에 담긴 물건을 선택하고 싶다면, `torch.gather()` 함수를 사용하여 선택할 수 있습니다. 다음과 같이 사용할 수 있습니다:
items = ['사과', '바나나', '딸기', '오렌지'] box_numbers = [2, 0, 1, 3] selected_item = torch.gather(items, 0, torch.tensor([2])) # 2번 상자의 물건을 선택
*0은 첫 번째 차원을 나타내며, 0은 텐서의 가장 왼쪽 차원입니다. 따라서 torch.gather() 함수는 첫 번째 차원에서 값을 수집합니다.
이 코드에서 `torch.gather(items, 0, torch.tensor([2]))`는 `items` 텐서에서 인덱스 2에 해당하는 값을 선택하여 최종적으로 `’딸기’`라는 값을 반환합니다. 즉, 2번 상자에 담긴 물건이 선택되었습니다.
이와 같이 `torch.gather()` 함수는 주어진 텐서에서 특정 인덱스 값을 선택하여 결과를 반환하는 기능을 수행합니다.