【PyTorch】PixelShuffle

PyTorch中的PixelShuffle_gdymind的博客-CSDN博客_nn.pixelshuffle

 

定义

该类定义如下：

class torch.nn.PixleShuffle(upscale_factor)

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

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

说白了就是使通道数降低，但是feature map变大，对于照片来说就是增加图像大小，降低通道数