Yogen想学算法的博客

原创 pytorch的逆亚像素卷积

由于Pytorch没有逆亚像素卷积，但是又临时需要，就参考亚像素卷积实现了一小段代码。 def de_pixelshuffle(input, downscale_factor): # channal batch_size, channels, in_height, in_width = input.size() out_height = in_height // downscale_factor out_width = in_width // downscale_factor