mindspore.ops.Rint

查看源文件
class mindspore.ops.Rint[源代码]

逐元素计算最接近输入数据的整数。

输入:
  • input_x (Tensor) - 任意维度输入Tensor,数据必须是float16、float32或者float64。

输出:

Tensor,shape和数据类型与 input_x 相同。

异常:
  • TypeError - 如果 input_x 的数据类型不是float16、float32、float64。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> input_x = Tensor(np.array([-1.6, -0.1, 1.5, 2.0]), mindspore.float32)
>>> op = ops.Rint()
>>> output = op(input_x)
>>> print(output)
[-2.  0.  2.  2.]
>>> input_x = Tensor(np.array([[-2.0, -1.9, -1.8, -1.7, -1.6],
...                            [-2.0, -1.9, -1.8, -1.7, -1.6]]), mindspore.float32)
>>> output = op(input_x)
>>> print(output)
[[-2. -2. -2. -2. -2.]
 [-2. -2. -2. -2. -2.]]