ML Katas

by: newest upvotes saves

Swap Height and Width

this year easy (<10 mins) | einops tensor manipulation image processing

### Description For a batch of images or feature maps, swap the height and width dimensions. The input shape is `(B, C, H, W)` and the output shape should be `(B, C, W, H)`. ### Starter Code...

1