[ 仅 paddle 参数更多 ]torch.nn.ConstantPad2d

torch.nn.ConstantPad2d

torch.nn.ConstantPad2d(padding,
                       value)

paddle.nn.Pad2D

paddle.nn.Pad2D(padding,
                mode='constant',
                value=0.0,
                data_format='NCHW',
                name=None)

其中 Paddle 和 PyTorch 的 padding 参数所支持的类型不一致，具体如下：

参数映射

PyTorch PaddlePaddle 备注
padding padding 填充大小，PyTorch 和 Paddle 的 padding 参数的类型分别为 (int/tuple) 和 (int/Tensor/list)。
value value 以 'constant' 模式填充区域时填充的值。默认值为 0.0 。
- mode padding 的四种模式，PyTorch 无此参数，Paddle 保持默认即可。
- data_format 输入和输出的数据格式，PyTorch 无此参数，Paddle 保持默认即可。

转写示例

padding：填充大小

# PyTorch 写法
m = nn.ConstantPad2d((3, 1), 3.5)
m(input)

# Paddle 写法
m = nn.Pad2D([3, 1], value=3.5)
m(input)