nn.PixelShuffle

class torch.nn.PixleShuffle(upscale_factor)
这里的upscale_factor就是放大的倍数,数据类型为int。

例子
>>> ps = nn.PixelShuffle(3)
>>> input = torch.tensor(1, 9, 4, 4)
>>> output = ps(input)
>>> print(output.size())
torch.Size([1, 1, 12, 12])

 

posted on 2021-03-14 21:41  cltt  阅读(813)  评论(0编辑  收藏  举报

导航