torch.nn.Linear
입력 데이터에 선형변환(Y = Wx + b)을 한 후 결과를 출력하는 Layer
사용법
torch.nn.Linear(in_features, out_features,
bias=True, device=None, dtype=None)
in_features
: 각 입력 샘플의 크기
out_features
: 각 출력 샘플의 크기
bias
: False로 설정하면 layer에서 추가 bias를 학습하지 않음 (기본 True)