torch.nn.ConvTranspose2d(in_channels, out_channels, kernel_size, stride=1, padding=0, output_padding=0, groups=1,
bias=True, dilation=1, padding_mode='zeros', device=None, dtype=None)
计算公式:
output = ( input - 1 ) * s + k - 2p
其中:output 是输出尺寸,input 是输入尺寸 , s是步长,k是卷积后大小,p是padding
如果希望特征图尺寸变为原来2倍,通道变为原来一半:
torch.nn.ConvTranspose2d(in_channels, in_channels // 2, 2, 2)