Does it think about to support torch.nn.PixelShuffle? It is widely used in Super-Resolution.

Question

wangqiim opened this issue 2 years ago · comments

wangqiim · Answer 1 · Sat Jul 02 2022 00:14:19 GMT+0800 (China Standard Time)

It looks like it's always going to be 0 Flops 😂

Ligeng Zhu · Answer 2 · Sat Jul 02 2022 22:28:44 GMT+0800 (China Standard Time)

I think it is already added? PixelShuffle operation does not contain any multiplications, thus should be zero macs.

wangqiim · Answer 3 · Sat Jul 02 2022 23:00:44 GMT+0800 (China Standard Time)

I don't think it is added. I just open a pull request.

Ligeng Zhu · Answer 4 · Sat Jul 02 2022 23:30:27 GMT+0800 (China Standard Time)

PR Merged. Thanks for the contribution!