Побудова U-Net: метод forward

Маючи визначені шари енкодера та декодера, ви можете реалізувати метод forward() для U-Net. Вхідні дані вже пропущено через енкодер. Однак вам потрібно визначити останній блок декодера.

Мета декодера — підвищити роздільність (upsample) карт ознак так, щоб його вихід мав ту саму висоту та ширину, що й вхідне зображення U‑Net. Це дасть змогу отримати семантичні маски на рівні пікселів.