Where is the actual implementation/source code for nn.PixelShuffle?

I can find just a docstring when I go to torch.nn.functional.pixel_shuffle. Where can I find the actual implementation?

The implementation can be found here.

@ptrblck Thanks a bunch!!

I cannot find the cpp file in the installed packages anywhere. I want to change the implementation locally. Where can I do that?

You won’t be able to locally change a C++ source file and see the changes without rebuilding it.
To apply changes to the PixelShuffle implementation, use this guide to build PyTorch from source, manipulate the file, and rebuild it.
This Contributing README might also be useful for incremental builds.

1 Like

Thanks a lot for helping out. Really appreciate the effort.