All Katas - ML Katas

by: newest upvotes saves

Pixel Unshuffle (Pixel to Channel)

this year medium (<10 mins) | einops super-resolution pixel shuffle

### Description This is another name for the space-to-depth operation, common in super-resolution models. It involves rearranging blocks of spatial data into the channel dimension. Given a tensor...
Pixel Shuffle (Channel to Pixel)

this year medium (<10 mins) | einops super-resolution pixel shuffle

### Description The inverse of pixel unshuffle, also known as depth-to-space. It is used to upscale an image by rearranging elements from the channel dimension into spatial blocks. Given a tensor...