PaddlePaddle

API 指南

 paddle / round

round

paddle. round ( x, name=None )

该OP将输入中的数值四舍五入到最接近的整数数值。

输入：
    x.shape = [4]
    x.data = [1.2, -0.9, 3.4, 0.9]

输出：
    out.shape = [4]
    out.data = [1., -1., 3., 1.]

参数：

  • x (Tensor) - 支持任意维度的Tensor。数据类型为float32，float64或float16。

  • name (str，可选) – 具体用法请参见 Name ，一般无需设置，默认值为None。

返回：返回类型为Tensor， 数据类型同输入一致。

代码示例：

import paddle

x = paddle.to_tensor([1.2, -0.9, 3.4, 0.9], dtype='float32')
result = paddle.round(x)
print(result) # result=[1., -1., 3., 1.]

此页内容是否对您有帮助

上一篇 下一篇
©Copyright 2011-2021 www.yanxishe.com ©AI研习社 All Rights Reserved 粤ICP备11095991号-21