[fix] Dimension out of range in pixel_shuffle / pixel_unshuffle (#54086)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/54051
Problem was application of the unary minus operator to an unsigned type. Positive indices are now used to build the permutation array for both `pixel_shuffle` and `pixel_unshuffle`.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/54086
Reviewed By: agolynski
Differential Revision: D27093435
Pulled By: jbschlosser
fbshipit-source-id: 4062f71277d037e91dc3cf5835b29b8ed4d16607