[Pytorch] gather 함수

이번에는 pytorch의 gather 함수에 대해 포스팅 해보도록 하겠습니다. 간단하게 torch.gather는 특정 인덱스를 쉽게 추출하기 위한 함수입니다! 다음은 파이토치 공식 문서에 있는 내용입니다. 공식문서의 내용이 궁금하신 분은 밑 함수부분에 링크를 연결해놓았으니 참조해주시기 바랍니다. torch.gather(input, dim, index, *, sparse_grad=False, out=None) → Tensor out[i][j][k] = input[index[i][j][k]][j][k] # if dim == 0 out[i][j][k] = input[i][index[i][j][k]][k] # if dim == 1 out[i][j][k] = input[i][j][index[i][j][k]] # ..

2023.04.12