[ torch 参数更多 ] torch.Tensor.div_¶

torch.Tensor.div_¶

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

paddle.Tensor.divide_¶

paddle.Tensor.divide_(y)

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

参数映射¶

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

转写示例¶

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