
PixelUnshuffle函数的手动实现
众所周知,PixelUnshuffle函数是深度学习中一个用于图像或特征图操作的函数,其核心作用是将输入张量中的像素块进行重新排列,把原本空间维度(高度和宽度)上的信息转移到通道维度上,从而减少空间维度大小,增加通道维度大小。如下图所示,可以将一个(1,1,4,4)的张量变为(1,4,2,2)。现在开始观察,我们可以发现,要做的其实就是将一个4*4的图片变为四个2*2的图片,也就是将输入特征图的分辨率降低。此时,张量的通道数变成了:channels*r*r,高度与宽度分别变成了之前的一半。









