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

torch.Tensor.round_

torch.Tensor.round_(decimals=0)

paddle.Tensor.round_

paddle.Tensor.round_(name=None)

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

参数映射

PyTorch PaddlePaddle 备注
decimals - 舍入小数位数

转写示例

decimals:要舍入到的小数位数

# PyTorch 写法
torch.tensor([3.345, 5.774]).round_(decimals=2)

# Paddle 写法
(paddle.to_tensor([3.345, 5.774]) * 1e2).round_() / 1e2

# 注:Paddle 可使用 10 的 decimals 次方来实现