nn.PixelShuffle

class torch.nn.PixleShuffle(upscale_factor)
这里的upscale_factor就是放大的倍数，数据类型为int。

例子
>>> ps = nn.PixelShuffle(3)
>>> input = torch.tensor(1, 9, 4, 4)
>>> output = ps(input)
>>> print(output.size())
torch.Size([1, 1, 12, 12])