torch.nn.functional.pixel_unshuffle¶
- torch.nn.functional.pixel_unshuffle(input, downscale_factor) 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])