14 pytorch.nn.ConvTranspose2d类输出尺寸计算方法

torch.nn.ConvTranspose2d类输出尺寸计算方法

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')
inchannels = 64
out_channels = 3
kernel_size = 5
stride=3
padding=1
output = (input-1)stride+outputpadding -2padding+kernelsize

你可能感兴趣的:(14 pytorch.nn.ConvTranspose2d类输出尺寸计算方法)