Data/Data Science
2021. 5. 3.
[Pytorch] torch.nn layer 함수 정리 (추가정리중)
눈으로만 봐서는 도저히 머리에 안들어와서 블로그에 쓰면서 정리함 pytorch.org/docs/stable/nn.html torch.nn — PyTorch 1.8.1 documentation Shortcuts pytorch.org torch.nn.Linear - 함수 선언시 두개의 인자값이 들어가게되고, 첫번째 인자값은 input size, 두번째 인자값은 output size이다 - 실제로 데이터가 거치는 forward()부분에선 두번째 인자값없이 첫번째 인자값만 넘겨주면된다. return되는 tensor의 크기가 output size가 된다 # Linear layer 선언 linear_1 = torch.nn.Linear(input_size, output_size) # forward 부분 (실제로 데이..
![[Pytorch] torch.nn layer 함수 정리 (추가정리중)](http://i1.daumcdn.net/thumb/C480x320/?fname=https://blog.kakaocdn.net/dn/Ht8xK/btq3Ucj6Ien/qkN1zqoGb8h6NMWkdOA9g0/img.png)