PyTorch中的PixelShuffle

感性认识

一般的convolution操作会使feature map变小，
但当我们的$\text{stride}=\frac{1}{r}<1$时，可以让卷积后的feature map变大，这个新的操作叫做sub-pixel convolution，具体原理可以看Real-Time Single Image and Video Super-Resolution Using an Efficient Sub-Pixel Convolutional Neural Network这篇paper。

定义

该类定义如下：

class torch.nn.PixleShuffle(upscale_factor)

这里的upscale_factor就是放大的倍数。

输入输出的shape

具体一点来说，Pixelshuffle会将shape为 ( ∗ , r 2