pytorch的pixelshuffle与tensorflow的depth2space的完美等价替换
例如 C=48, shuffle 上采样4x4
将pytorch nn.PixelShuffle(4) 的输入前的卷积权重和偏置进行通道上的重排即可
if v.shape[0] == 48:
norder = np.arange(48).reshape([-1, 4*4]).transpose(1, 0).reshape([-1]).tolist()
v = v[norder, ...]