Request: PixelShuffle - Githubissues

huggingface / candle

Minimalist ML framework for Rust

Apache License 2.0

13.79k stars 751 forks source link

Open oovm opened 3 weeks ago

oovm commented 3 weeks ago

I hope to add PixelShuffle and PixelUnshuffle in pytorch

I found this layer missing when porting some super-resolution and GAN models

LaurentMazare commented 3 weeks ago

Wouldn't the functions in candle_nn::ops work for you? Here is the pixel_shuffle one.