torch.nn.functional.pixel_unshuffle¶
- torch.nn.functional.pixel_unshuffle(input, downscale_factor) Tensor ¶
通过重新排列形状为 的 Tensor 元素,将其反转为形状为 的 Tensor,该操作是
PixelShuffle
操作的逆向过程,其中 r 是downscale_factor
。详情请参阅
PixelUnshuffle
。示例
>>> input = torch.randn(1, 1, 12, 12) >>> output = torch.nn.functional.pixel_unshuffle(input, 3) >>> print(output.size()) torch.Size([1, 9, 4, 4])