본문 바로가기

Data Science/PyTorch

(3)
torch 공식 문서와 친해지기 https://pytorch.org/docs/stable/torch.html torch — PyTorch 1.12 documentation Shortcuts pytorch.org PyTorch를 처음 시작하거나, 몇몇 함수의 사용법이 헷갈린다면 이 문서를 참조하는 것이 큰 도움이 될 것 같다. tensor를 능숙하게 다루기 위한 필수적인 함수들이 잘 정리되어 있다. 문서를 천천히 읽어 내려가보면, 이런 목차들을 만날 수 있다. Creation Ops Indexing Slicing Joining Mutating Ops Random Sampling Math operations 딱 봐도 "중요해보인다...!" 라는 생각이 절로 든다. 여기 나오는 함수들은 잘 익혀두는게 신상에 이로울 것 같다.
torch.Tensor / torch.tensor https://pytorch.org/docs/stable/generated/torch.tensor.html torch.tensor — PyTorch 1.12 documentation Shortcuts pytorch.org 코드 실행해보기 import torch ex_list = [1, 2 ,3] T_ex = torch.Tensor(ex_list) t_ex = torch.tensor(ex_list) ex_list[0] = 4 print(T_ex, t_ex) print('-' * 40) ex_tensor = torch.tensor(t_ex) # t_ex = tensor([1, 2, 3]) T_ex = torch.Tensor(ex_tensor) t_ex = torch.tensor(ex_tensor) ex_te..
torch.gather / torch.scatter_ https://pytorch.org/docs/stable/generated/torch.gather.html?highlight=gather#torch.gather torch.Tensor.scatter_ — PyTorch 1.12 documentation Shortcuts pytorch.org 정말 헷갈리는 torch.gather 함수... 제대로 이해하기 위해 여러 자료를 검색해봤지만, 공식 문서에 나와있는 아래의 수식을 이해하는 것이 가장 빠르고 정확한 길이란걸 깨달았다. https://pytorch.org/docs/stable/generated/torch.Tensor.scatter_.html#torch.Tensor.scatter_ torch.Tensor.scatter_ — PyTorch 1.12 doc..