일반적인 PyTorch 모델
1. 모듈은 중첩될 수 있으며 torch.nn.Module 에서 상속됨.
2. 모델은 계층을 인스턴스화하고 필요한 데이터 아티팩트를 로드하는 __init__() 함수를 가지고 있다.
3. 모델은 forward() 함수를 가지고 있고 여기서 실제 계산이 수행된다. 입력은 네트워크 계층 및 다양한 기능들을 통과시켜 결과를 생성한다.
4. 그 외에는 다른 파이썬 클래스처럼 모델 클래스를 구성할 수 있다.
input = torch.rand(1, 1, 32, 32)
→ 배치 크기(batch size) = 1, 이미지의 채널(channel) 수 = 1, 이미지의 높이(height) = 32, 이미지의 너비(width) = 32
→ 이때 LeNet을 통과한 output shape = [1(배치 크기(batch size)), 10(각 클래스에 대한 출력값)]