Enhance transpose operators
Summary: This commit allows transpose to compute in-place by leveraging buffer. We also adds CRD mode for space-depth transpose (i.e., pixel shuffle).
Showing
with
757 additions
and
231 deletions
docs/api/python/torch/nn/PixelShuffle.rst
0 → 100644
docs/api/python/torch/nn/PixelUnshuffle.rst
0 → 100644
torch/core/nn/modules/pixelshuffle.py
0 → 100644
-
Please register or sign in to post a comment