Add function signature for pixel_shuffle (#45661)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/45661
Test Plan: Imported from OSS
Reviewed By: gmagogsfm
Differential Revision: D24078627
Pulled By: ansleyadelaide
fbshipit-source-id: 44917ff5932e4d0adcc18ce24ecfc0b5686818e3