https://pajamacoder.tistory.com/2
[pytorch] torch.gather 설명
pytorch 코드를 분석하다 torch.gather() 메서드의 동작이 헷갈려 정리해 본다. torch.gather() 메서드는 input tensor로 부터 원하는 차원에서 주어진 index에 해당하는 원소만을 골라 새로운 텐서를 만들때 사
pajamacoder.tistory.com
'자연어 처리 과정' 카테고리의 다른 글
| What is Learning Rate Warmup? (1) | 2023.01.16 |
|---|---|
| Learning Rate Schedulers(PyTorch) (0) | 2023.01.16 |
| Confusion matrix (0) | 2023.01.13 |
| pytorch의 manual_seed (0) | 2023.01.13 |
| What's the role of "pooler" in BERT? (0) | 2023.01.11 |
