candle icon indicating copy to clipboard operation
candle copied to clipboard

Published 20 hours ago • verified publisher icon huggingface

Request: PixelShuffle

Open oovm opened this issue 1 year ago • 1 comments

I hope to add PixelShuffle and PixelUnshuffle in pytorch

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

oovm avatar Apr 21 '24 13:04 oovm

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

LaurentMazare avatar Apr 21 '24 16:04 LaurentMazare