<img src="https://cdn.inflearn.com/public/files/posts/091d3e9a-b6c1-486c-a2e7-36936d5dcf38/blob" alt="" />
self.fc1 =nn.Linear(320,50)에서 320이란 숫자가 어떻게 나온건지 모르겠습니다.
(W-F+2P)/S + 1을 적용해보면
1) (28-5+2*0)/1 + 1 =&gt; 24 2) (24-5+2*0)/1 + 1 =&gt; 20 3) 20 x 20 x 20(output depth) =&gt; 8,000이라고 생각했습니다.

안녕하세요, Justin입니다.
(W - F + 2P) / S + 1 을 이용해서 Output shape을 잘 계산해주셨찌만, F.max_pool2d를 생략하신 것 같습니다.
1) 결과에서 F.max_pool2d를 이용하면 (12 * 12 * 10) 크기로 출력되고, 2) 결과에서 F.max_pool2d를 이요하면 (4 * 4 * 20) 크기로 출력됩니다.
따라서 Flatten되는 최종 개수는 4 * 4 * 20으로 320입니다.
감사합니다.

인프런 커뮤니티 질문&답변

CNN_MNIST 실습예제 질문입니다.