Does it think about to support torch.nn.PixelShuffle? It is widely used in Super-Resolution.
wangqiim opened this issue · comments
It looks like it's always going to be 0 Flops 😂
I think it is already added? PixelShuffle operation does not contain any multiplications, thus should be zero macs.
I don't think it is added. I just open a pull request.
PR Merged. Thanks for the contribution!