round
round
(x, name=None)
- import numpy as np
- inputs = fluid.layers.data(name="x", shape = [3], dtype='float32')
- output = fluid.layers.round(inputs)
- exe = fluid.Executor(fluid.CPUPlace())
- 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)