[ 仅 paddle 参数更多 ]torch.nn.functional.pad

torch.nn.functional.pad

torch.nn.functional.pad(input,
                            pad,
                            mode='constant',
                            value=None)

paddle.nn.functional.pad

paddle.nn.functional.pad(x,
                            pad,
                            mode='constant',
                            value=0.0,
                            data_format='NCHW',
                            name=None)

两者功能一致,其中 Paddle 相比 PyTorch 支持更多其他参数,具体如下:

参数映射

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | input | x | 表示输入的 Tensor,仅参数名不一致。 | | pad | pad | 表示一个 one-hot 向量的长度 。 | | mode | mode | 表示填充的模式。 | | value | value | 表示填充的值,mode 为’constant’时有效 。 | | - | data_format | 指定输入的数据格式, PyTorch 无此参数, Paddle 保持默认即可。 |

在实际使用过程中,data_format 参数需要根据输入参数进行指定