[ 输入参数类型不一致 ]torch.nn.ReplicationPad2d

torch.nn.ReplicationPad2d

torch.nn.ReplicationPad2d(padding)

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) ,需要转写。
- mode padding 的四种模式,PyTorch 无此参数,Paddle 需设置为replicate
- value 表示填充值,PyTorch 无此参数,Paddle 保持默认即可。
- data_format 输入和输出的数据格式,PyTorch 无此参数,Paddle 保持默认即可。

转写示例

padding:填充大小

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

# Paddle 写法
pad = paddle.to_tensor((3, 1))
m = nn.Pad2D(pad, mode='replicate')
m(input)