Debug a PyTorch U-Net shape mismatch

Q: How do I practice coding and algorithm questions?

Use PracHub's coding console to write, test, and debug your solutions in Python or JavaScript. View hints, test against sample inputs, and compare with official solutions.

Q: What difficulty level is this coding question?

This is a medium difficulty Coding & Algorithms question, commonly asked during Technical Screen rounds at Apple.

Q: What role is this question designed for?

This question is commonly asked for Machine Learning Engineer candidates at Apple during technical interviews.

Question

Debug a PyTorch U-Net shape mismatch

PracHub · Accepted Answer

This question evaluates proficiency in PyTorch model implementation, convolutional output shape arithmetic, channel configuration, and debugging of U-Net-style segmentation networks with valid (no-padding) convolutions.

Quick Overview

Debug a PyTorch U-Net shape mismatch