torch.nn.functional.pixel_shuffle¶
- torch.nn.functional.pixel_shuffle(input, upscale_factor) Tensor ¶
将形状为 的张量中的元素重新排列到形状为 的张量中,其中 r 是
upscale_factor
。有关详细信息,请参见
PixelShuffle
。示例
>>> input = torch.randn(1, 9, 4, 4) >>> output = torch.nn.functional.pixel_shuffle(input, 3) >>> print(output.size()) torch.Size([1, 1, 12, 12])