[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 次方来实现