round

  • paddle.fluid.layers.round(x, name=None)

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

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

参数:

  • x (Variable) - 支持任意维度的Tensor。数据类型为float32，float64或float16。
  • name (str，可选) – 具体用法请参见 Name ，一般无需设置，默认值为None。

返回：返回类型为Variable(Tensor|LoDTensor)， 数据类型同输入一致。

代码示例：

  1. import numpy as np
  2. import paddle.fluid as fluid
  3.  
  4. inputs = fluid.layers.data(name="x", shape = [3], dtype='float32')
  5. output = fluid.layers.round(inputs)
  6.  
  7. exe = fluid.Executor(fluid.CPUPlace())
  8. exe.run(fluid.default_startup_program())
  9.  
  10. img = np.array([1.2, -0.9, 3.4, 0.9]).astype(np.float32)
  11.  
  12. res = exe.run(fluid.default_main_program(), feed={'x':img}, fetch_list=[output])
  13. print(res)
  14. # [array([ 1., -1.,  3.,  1.], dtype=float32)]