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