반응형 파이토치/신경망 레이어1 파이토치 nn.Linear 설명 nn.Linear 은 신경망 은닉층의 한 레이어를 만들어줍니다. 선형함수입니다. 두가지 인자가 입력되는데 하나는 입력텐서의 차원이고 다른 하나는 출력텐서의 차원입니다. 예를 들어 아래와 같은 레이어를 만들었다고 합시다. import torch.nn as nn layer1=nn.Linear(3,1) 입력 텐서는 세개의 차원을 갖습니다. $[x_{1},x_{2},x_{3}]$ 라고 쓰겠습니다. 출력텐서는 하나의 차원을 갖습니다. y라고 쓰겠습니다. 이때 생성된 layer 는 아래와 같은 형태의 선형 함수입니다. $y=w_{1}x_{1}+w_{2}x_{2}+w_{3}x_{3}+b$ b는 bias 입니다. nn.Linear 은 위와 같은 형태의 함수를 정의해줍니다. nn.Linear 에 담겨있는 정보는 weig.. 2023. 11. 3. 이전 1 다음 반응형