【Pytorch】nn.PixelShuffle

【Pytorch】nn.PixelShuffle

torch.nn.PixelShuffle(upscale_factor)

图片来源：[1]

输入输出尺寸：

import torchimport torch.nn as nnps = nn.PixelShuffle(2)input = torch.arange(0, 8 * 2 * 2).view(1, 8, 2, 2)output = ps(input)print('input:\n',input)print('output:\n',output)

[1] Real-Time Single Image and Video Super-Resolution Using an Efficient Sub-Pixel Convolutional Neural Network https://openaccess.thecvf.com/content_cvpr_2016/papers/Shi_Real-Time_Single_Image_CVPR_2016_paper.pdf

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。