nn.conv2d()

在用conv2d函数计算网络中间feature的大小时,发现不同博客提供的公式还不一样,后来发现是不同框架的区别。
tensorflow中,padding填充有same、valid两种模式,分别对应两种计算公式。
但是在pytorch中尺寸的计算是统一的:
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/qq_41872271/article/details/105662817