nn.PixelShuffle
摘要:
class torch.nn.PixleShuffle(upscale_factor) 这里的upscale_factor就是放大的倍数,数据类型为int。 例子 >>> ps = nn.PixelShuffle(3) >>> input = torch.tensor(1, 9, 4, 4) >>> 阅读全文
posted @ 2021-03-14 21:41 cltt 阅读(813) 评论(0) 推荐(0) 编辑