torch.ConvTranspose2d的计算
转置卷积实际是根据输入卷积核和输入特征图生成中间卷积核和特征图,再进行卷积操作
输入的卷积核kernel 大小 Size = size 步长 Stride = stride 填充 Padding = padding
输入特征图 H W
中间卷积核kernel' 大小 Size' = size 步长 Stride' = 1 填充 Padding' = size - padding - 1
中间特征图 H' = H + (stride-1) * (H-1)
输出特征图 H'' = (H' + 2*Padding' - Size') / Stride' +1
= stride (H - 1) + size -2*padding