[ 参数不一致 ]torch.narrow¶
torch.narrow¶
torch.narrow(input,
dim,
start,
length)
paddle.slice¶
paddle.slice(input,
axes,
starts,
ends)
其中 Pytorch 的 length 与 Paddle 的 ends 用法不一致,具体如下:
参数映射¶
| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | input | input | 表示输入的 Tensor 。 | | dim | axes | 表示切片的轴。 | | start | starts | 表示起始位置。 | | length | - | 到结束位置的长度,Paddle 无此参数。应修改 ends 实现。 | | - | ends | 表示结束位置,Pytorch 无此参数。 Paddle 应设为 start + length。 |
转写示例¶
# PyTorch 写法:
torch.narrow(x, 0, 1, 2)
# Paddle 写法:
# Paddle 可通过设置 ends-starts=length 来实现 Pytorch 的 length 功能
paddle.slice(x, [0], [1], [3])