round
paddle.fluid.layers.
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 (Variable) - 支持任意维度的Tensor。数据类型为float32,float64或float16。
- name (str,可选) – 具体用法请参见 Name ,一般无需设置,默认值为None。
返回:返回类型为Variable(Tensor|LoDTensor), 数据类型同输入一致。
代码示例:
- import numpy as np
- import paddle.fluid as fluid
- inputs = fluid.layers.data(name="x", shape = [3], dtype='float32')
- output = fluid.layers.round(inputs)
- exe = fluid.Executor(fluid.CPUPlace())
- exe.run(fluid.default_startup_program())
- img = np.array([1.2, -0.9, 3.4, 0.9]).astype(np.float32)
- res = exe.run(fluid.default_main_program(), feed={'x':img}, fetch_list=[output])
- print(res)
- # [array([ 1., -1., 3., 1.], dtype=float32)]