torch.gather / torch.scatter_

https://pytorch.org/docs/stable/generated/torch.gather.html?highlight=gather#torch.gather

torch.Tensor.scatter_ — PyTorch 1.12 documentation

 

 

 

정말 헷갈리는 torch.gather 함수...

 

 

제대로 이해하기 위해 여러 자료를 검색해봤지만,

 

공식 문서에 나와있는 아래의 수식을 이해하는 것이

 

가장 빠르고 정확한 길이란걸 깨달았다.

 

 

 

---

https://pytorch.org/docs/stable/generated/torch.Tensor.scatter_.html#torch.Tensor.scatter_

torch.Tensor.scatter_ — PyTorch 1.12 documentation

 

 

 

torch.scatter_ 함수도 마찬가지!

 

공식문서