ConvTranspose2d尺寸计算
时间: 2023-12-11 15:30:08 浏览: 84
nn.ConvTranspose2d 逆卷积 反卷积
ConvTranspose2d是一种反卷积操作,用于将低维特征图扩展到高维特征图。在计算ConvTranspose2d的输出尺寸时,需要考虑输入特征图的大小、卷积核大小、步长、边界扩充等因素。具体计算公式如下:
输出特征图大小 = (输入特征图大小 - 1) * 步长 + 卷积核大小 - 2 * 边界扩充
其中,输入特征图大小、卷积核大小、步长、边界扩充均为引用中提到的参数。需要注意的是,ConvTranspose2d的步长默认为1,边界扩充默认为0。如果有输出填充,还需要加上out_padding的大小。
阅读全文