[PyTorch] nn.Linear

pytorch에서 선형회귀 모델은 nn.Linear() 함수에 구현되어 있다. 입력되는 x의 차원과 출력되는 y의 차원을 입력해 주면 된다. 단순 선형회귀는 하나의 입력 x에 대해 하나의 입력 y가 나오니 로 하면 된다. PyTorch 공식 문서 내용을 보면 으로 나와있다. Parameter를 보면 in_features는 input sample의 size out_features는 output sample의 size bias는 만일 False로 설정되어 있으면 layer는 bias를 학습하지 않는다. 기본값은 True이다. device는 CPU, GPU 중 고르는 거고 dtype은 자료형의 타입을 정하는 것 같다. Shape input에서 *은 어떤 additional dimension을 의미하고 H(in)은 in_features를 의미한다. Ouput 마지막 차원을 제외..........

원문링크 : [PyTorch] nn.Linear