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