pytorch 等效于 Tenserflow 中的 Conv2D，步幅为 2，填充为 (1,1)

青葱年少 2年前 pytorch 203

原文标题 ：pytorch equivalent of Conv2D in tenserflow with stride of 2 and padding of (1,1)

我有conv1 = nn.Conv2d(3, 16, 3,stride= 2, padding = 1, bias=True, groups=1)。我需要它对应的 api intf.keras.layers.Conv2D。

谁能帮我吗

PS：这里我大步2

原文链接：https://stackoverflow.com//questions/71979310/pytorch-equivalent-of-conv2d-in-tenserflow-with-stride-of-2-and-padding-of-1-1

我来回复

Prajot Kuvalekar 评论

我找到了解决方案，希望这对其他人也有帮助。因为很难匹配paddingintorch和paddinginkeras和stride = 2

X = Input(shape = (10,10,3))
X1 = ZeroPadding2D(padding=(1,1), input_shape=(10, 10, 3), data_format="channels_last")(X)
conv1 = Conv2D(16, 3,  padding = 'valid', strides = (2,2))(X1)

2年前 0条评论

pytorch 等效于 Tenserflow 中的 Conv2D，步幅为 2，填充为 (1,1)

回复

相关问题