자연어 처리 과정 torch.gather chanuu 2023. 1. 16. 16:30 https://pajamacoder.tistory.com/2 [pytorch] torch.gather 설명 pytorch 코드를 분석하다 torch.gather() 메서드의 동작이 헷갈려 정리해 본다. torch.gather() 메서드는 input tensor로 부터 원하는 차원에서 주어진 index에 해당하는 원소만을 골라 새로운 텐서를 만들때 사 pajamacoder.tistory.com
