파이토치 선형변환

Deep Learning

Pytorch의 nn.Linear() 파헤치기

Pytorch를 이용해 짠 대부분의 소스코드들에는 nn.Linear() 함수가 거의 꼭 들어가 있는 것을 알 수 있을 것이다. 그만큼 자주 많이 사용되고 크게 어려운 개념은 아니지만, 그렇다고 대충 넘어가면 추후 코드를 이해함에 있어서 문제가 생길 수 있다. 따라서 nn.Linear()에 대해 차근차근 파헤쳐보자. nn.Linear()란? Pytorch에서 선형회귀 모델은 nn.Linear() 함수에 구현되어 있다. Pytorch 공식 문서에서 설명하는 nn.Linear() 함수는 다음과 같다. 입력 인자로 in_features와 out_features를 받고 이에 맞춰 반환한다. 예를 들어 내가 768차원 짜리를 64차원으로 만들고 싶으면 nn.Linear(768,64) 위와 같은 형태로 작성하면 된다..

seungseop
'파이토치 선형변환' 태그의 글 목록