U-Net: Concatenating Skip Connections

The problem, from first principles

The skip connection in U-Net concatenates the encoder's features with the upsampled decoder features. The decoder then has both coarse context and fine spatial detail for the next convolution.

From theory to code

Implement skip_concat(enc, dec), joining the two feature maps along the channel axis.

Constraints

• Both maps have the same height and width.

Hints