U-Net 구축: 레이어 정의

이번 연습 문제와 다음 연습 문제에서는 U-Net 아키텍처를 처음부터 구성해 보겠습니다. 여기서는 모델이 사용할 레이어와 레이어 블록을 정의하는 __init__() 메서드부터 시작합니다.

인코더와 전치 합성곱은 이미 정의되어 있습니다. 남은 것은 디코더의 합성곱 블록입니다. 스킵 연결을 고려하여 각 블록에 적절한 입력 및 출력 채널 수를 전달해야 합니다.

첫 번째 블록인 dec1은 upconv3의 출력과 enc3의 출력을 이어 붙인(concatenation) 것을 입력으로 받습니다. 그리고 dec1의 출력은 enc3의 출력과 동일해야 합니다. 누락된 입력과 출력 크기를 모두 채워 넣을 수 있을까요?