pytorch
4fac72ee - [fix] Dimension out of range in pixel_shuffle / pixel_unshuffle (#54086)

Commit
3 years ago
[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
Author
Parents
Loading