[ torch 参数更多 ]torch.Tensor.sub_

torch.Tensor.sub_

torch.Tensor.sub_(other, *, alpha=1)

paddle.Tensor.subtract_

paddle.Tensor.subtract_(y)

Pytorch 相比 Paddle 支持更多其他参数,具体如下:

参数映射

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | other | y | 表示减数的 Tensor,仅参数名不一致。 | | alpha | - | 表示other的乘数,Paddle 无此参数,需要转写。Paddle 应设置 y = alpha * other。 |

转写示例

alpha:表示other的乘数

# Pytorch 写法
x.sub_(y, alpha=2)

# Paddle 写法
x.subtract_(2 * y)

# 注:Paddle 直接将 alpha 与 y 相乘实现