← 보드nn.Module로 모델 만들기 · 8/8

이 장 3줄 요약

모델은 nn.Module을 물려받은 클래스예요. __init__에서 super().__init__()을 맨 먼저 쓰고 층을 만들며, forward에서 계산 흐름을 써요.
nn.Linear(입력, 출력)은 x @ W.T + B를 해 주는 층이에요. weight는 (출력, 입력), bias는 (출력,), 파라미터 개수는 입력×출력+출력이고, 출력은 (배치, 출력)이에요.
model(x)를 쓰면 부모의 __call__을 거쳐 forward가 실행돼요. __init__은 모델을 만들 때 한 번뿐이고, 일반 Tensor가 아니라 nn.Parameter(또는 nn.Linear)여야 파라미터로 등록돼요.

자료: 3·6주차 · Week03_Class_for_PyTorch(업데이트), Week04_Autograd_Loss_Optimizer · 6주차 녹취(빅데이터06), 7주차 녹취(빅데이터07)