Debug a PyTorch U-Net shape mismatch

Company: Apple

Role: Machine Learning Engineer

Category: Coding & Algorithms

Difficulty: medium

Interview Round: Technical Screen

Overview: 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.

Read the full Apple Machine Learning Engineer interview experience this question came from