[ 参数不一致 ]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])