关于具体内容:
解释超级好!可视化
如果只是想拿来当作一个工具的话 根本不要考虑那么多,
如果保持
nn.conv2d跟nn.convTranspose2d中的参数一致,那么就是input 跟output互相转换,如下:
input1 = torch.randn(4,3,4,4)
input2 = torch.randn(4,3,2,2)
print("-------------------------down---------------------------")
m1 = nn.Conv2d(3,8,3,stride=1,padding=0