mindspore.ops.Rint
- class mindspore.ops.Rint[源代码]
逐元素计算最接近输入数据的整数。
- 输入:
input_x (Tensor) - 待计算的Tensor,数据必须是float16、float32或者float64。shape为 \((N,*)\) ,其中 \(*\) 表示任意数量的附加维度。
- 输出:
Tensor,shape和数据类型与 input_x 相同。
- 异常:
TypeError - 如果 input_x 的数据类型不是float16、float32、float64。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> 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.]]