pytorch 等效于张量流中的 Conv2D，步幅为 2，填充为 (1,1)
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)。我需要 tf.keras.layers.Conv2D.
中对应的 api
谁能帮帮我
PS : 我的步幅是 2
我找到了解决办法，希望这对其他人也有帮助。因为很难将 torch 中的 padding 和 keras 中的 padding 与 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)
我有conv1 = nn.Conv2d(3, 16, 3,stride= 2, padding = 1, bias=True, groups=1)。我需要 tf.keras.layers.Conv2D.
谁能帮帮我
PS : 我的步幅是 2
我找到了解决办法，希望这对其他人也有帮助。因为很难将 torch 中的 padding 和 keras 中的 padding 与 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)