[ 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() # 向零取整