输入尺寸为 (1,2,2,2)
x = torch.rand(1,1, 2, 2)
f = nn.ConvTranspose2d(in_channels=1, out_channels=1, stride=2, kernel_size= 3, padding=0)
f(x).shape
运行得到的尺寸为:(1,1,5,5)
x = torch.rand(1,1, 2, 2)
f = nn.ConvTranspose2d(in_channels=1, out_channels=1, stride=3, kernel_size= 3, padding=0)
f(x).shape
同理运行上面的代码可以得到的尺寸为:(1,1,6,6)