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

torch.Tensor.div

torch.Tensor.div(other, *, rounding_mode=None)

paddle.Tensor.divide

paddle.Tensor.divide(y, name=None)

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

参数映射

PyTorch PaddlePaddle 备注
other y 表示输入的 Tensor ,仅参数名不一致。
rounding_mode - 用于指定在执行截断除法时的舍入模式,可选值为 'floor'(向下取整) 或 'trunc'(向零取整)。 Paddle 无此参数,需要转写。Paddle 可通过组合 paddle.trunc 或 paddle.floor 实现。

转写示例

# torch 写法
x = torch.tensor([4, 8, 12], dtype=torch.float32)
y = torch.tensor([3, 4, 5], dtype=torch.float32)
z1 = x.div(y, rounding_mode='floor') # 向下取整
z2 = x.div(y, rounding_mode='trunc') # 向零取整

# paddle 写法
x = paddle.to_tensor([4, 8, 12], dtype='float32')
y = paddle.to_tensor([3, 4, 5], dtype='float32')
z1 = x.divide(y).floor()  # 向下取整
z2 = x.divide(y).trunc()  # 向零取整