[ 输入参数类型不一致 ]torch.Tensor.slice_scatter

torch.Tensor.slice_scatter

torch.Tensor.slice_scatter(src, dim=0, start=None, end=None, step=1)

paddle.Tensor.slice_scatter

paddle.Tensor.slice_scatter(value, axes, starts, ends, strides, name=None)

两者功能一致,参数不一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
src value 嵌入的值,仅参数名不一致。
dim axes 嵌入的维度,PyTorch 为 int 类型,Paddle 为 list of int。
start starts 嵌入起始索引,PyTorch 为 int 类型,Paddle 为 list of int。
end ends 嵌入截至索引,PyTorch 为 int 类型,Paddle 为 list of int。
step strides 嵌入步长,PyTorch 为 int 类型,Paddle 为 list of int。

转写示例

# PyTorch 写法
x.slice_scatter(src, dim=0, start=1, end=5, step=2)

# Paddle 写法
x.slice_scatter(value, axes=[0], starts=[1], ends=[5], strides=[2])