[ 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=None,
                            pad_from_left_axis=True,
                            name=None)

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

参数映射

PyTorch PaddlePaddle 备注
input x 表示输入的 Tensor，仅参数名不一致。
pad pad 表示一个 one-hot 向量的长度 。
mode mode 表示填充的模式。
value value 表示填充的值，mode 为'constant'时有效 。
- data_format 指定输入的数据格式, PyTorch 无此参数， Paddle 保持默认即可。
- pad_from_left_axis 只有当 mode 为 'constant' ，且 pad 是长度为 2N 的列表时有效，设置 pad 与 x 的轴左对齐或右对齐， PyTorch 无此参数， Paddle 需设置为 False 结果才与 PyTorch 一致。

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