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