[ torch 参数更多 ] torch.Tensor.div¶
torch.Tensor.div¶
torch.Tensor.div(other, *, rounding_mode=None)
参数映射¶
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() # 向零取整