round

paddle. round ( x, name=None )

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

  1. 输入:
  2. x.shape = [4]
  3. x.data = [1.2, -0.9, 3.4, 0.9]
  4. 输出:
  5. out.shape = [4]
  6. out.data = [1., -1., 3., 1.]

参数:

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

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

返回:返回类型为Tensor, 数据类型同输入一致。

代码示例

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