PyTorch中的PixelShuffle（）

PyTorch中的PixelShuffle（）

卷积操作会使得图像的尺寸变小
但是当我们的stride = 1/r<1时，可以让卷积后的图像的尺寸变大。
参考文章
该类定义如下：

class torch.nn.PixleShuffle(upscale_factor)  #upscale_factor指放大的倍数

输入输出的shape

N代表batch size

参考代码：

ps = nn.PixelShuffle(3)# 缩放到三倍，r == 3
input = torch.tensor(1, 9, 4, 4)## r^2 C == 9C == 9，所以C == 1
output = ps(input)
print(output.size())
# 输出为：
# torch.Size([1, 1, 12, 12])