2021年3月14日

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) 编辑

F.unfold F.fold F.normalize()

摘要: import torch.nn.functional as F F.unfold https://blog.csdn.net/qq_34914551/article/details/102940368 F.normalize()https://blog.csdn.net/ECNU_LZJ/artic 阅读全文

posted @ 2021-03-14 15:11 cltt 阅读(1171) 评论(0) 推荐(0) 编辑

导航