[torch 参数更多 ]torch.subtract

torch.subtract

torch.subtract(input,
               other,
               *,
               alpha=1,
               out=None)

paddle.subtract

paddle.subtract(x,
                y,
                name=None)

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

参数映射

PyTorch PaddlePaddle 备注
input x 表示被减数的 Tensor,仅参数名不一致。
other y 表示减数的 Tensor,仅参数名不一致。
alpha - 表示other的乘数,Paddle 无此参数。Paddle 应设置 y = alpha * other。
out - 表示输出的 Tensor,Paddle 无此参数。

功能差异

转写示例

alpha:表示other的乘数

# PyTorch 写法
torch.subtract(x, y, alpha=2)

# Paddle 写法
paddle.subtract(x, 2*y)

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

out:指定输出

# PyTorch 写法
torch.subtract(x, y, out=z)

# Paddle 写法
paddle.assign(paddle.subtract(x, y), z)