PyTorch: torch.nn.Linear

danbibibi·2022년 1월 26일
0

PyTorch 🔥

목록 보기
9/20

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)

profile
블로그 이전) https://danbibibi.tistory.com

0개의 댓글