当pad_mode = 'same'时,官网上下图,是指输出和输入的大小一样的意思吗?
可是我实践的结果却不是,输入224*224:
代码如下:
#创建测试数据x
from mindspore import Tensor
from mindspore import numpy
import mindspore
import mindspore.nn as nn
x = Tensor(np.ones([1,3,224, 224]))
conv1 = nn.Conv2d(3, 64, 7, 2)
x_conv1 = conv1(x)
结果确实112*112,如下图:
参数 stride 的数值会影响输出的 shape 。把 stride 从 2 改成 1,输出的shape 就会是(1, 64, 224, 224)