[torch 参数更多]torch.special.round¶

torch.special.round ¶

          # torch.special.round 为 torch.round 别名，参数和 torch.round 相同
torch.special.round(input, *, decimals=0, out=None)

          paddle.round(x, name=None)

         

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

PyTorch	PaddlePaddle	备注
input	x	输入的 Tensor，仅参数名不一致。
decimals	-	舍入到的小数位数，PaddlePaddle 无此参数，需要转写。
out	-	表示输出的 Tensor，Paddle 无此参数，需要转写。

           # PyTorch 写法
torch.special.round(x, decimals=2)

# Paddle 写法
paddle.round(1e2 * x) / 1e2

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

           # PyTorch 写法:
torch.special.round(x, out=y)

# Paddle 写法:
paddle.assign(paddle.round(x), y)