由于pytorch没有自动计算卷积层和池化层之后的图像大小,所有连接层Linear都需要自己计算图像大小,因此必须记住这些表达式。
卷积层W :行列宽度,h :行列高度。
)卷积核的宽度和高度
p :填充(填充0的个数
)卷积核的个数
s :步长
于是,我
wIdth=(wf2p )/S 1width=(w-f2p )/S 1width=(w-f2p )/s1
hIght=(hf2p )/S 1height=(h-f2p )/S 1height=(h-f2p )/s1
3358www.Sina.com/wIdth=(wf2p )/S 1width=(w-f2p )/S 1width=(w-f2p )/s1
hIght=(wf2p )/S 1height=(w-f2p )/S 1height=(w-f2p )/s1