본문 바로가기

linear

[Pytorch] torch.nn layer 함수 정리 (추가정리중)
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] Linear Regression Prediction
Data/Data Science 2020. 3. 11. [Pytorch] Linear Regression Prediction 간단한 Linear Regression모델 선언 import torch.nn as nn class LR(nn.Module): def __init__(self, in_size, output_size): super(LR, self).__init__() self.linear = nn.Linear(in_size, output_size) def forward(self, x): out = self.linear(x) return out # Linear Regression model 생성 model = LR(1, 1) # Linear Regression bias, weight 변수 확인 model.state_dict() """ OrderedDict([('linear.weight', tensor([[-0.3027]]))..

티스토리툴바