mindspore.mint.round

mindspore.mint.round(input, *, decimals=0)[源代码]

对输入数据进行四舍五入到最接近的整数数值。

\[out_i \approx input_i\]

说明

Ascend平台支持的输入数据类型包括bfloat16（Atlas训练系列产品不支持）、float16、float32、float64、int32、int64。

参数：

input (Tensor) - 输入Tensor。

关键字参数：

decimals (int, 可选) - 要舍入到的小数位数（默认值：0）。如果为负数，则指定小数点左侧的位数。支持输入单元素Tensor转换为int。 input 类型为int32或int64时， decimals 参数值必须为0。

返回：

Tensor，shape和数据类型与输入 input 相同。

异常：

TypeError - input 不是Tensor。
RuntimeError - input 为int32或int64类型时， decimals 值不为0。

支持平台：

Ascend

样例：

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, mint
>>> input = Tensor(np.array([0.8, 1.5, 2.3, 2.5, -4.5]), mindspore.float32)
>>> output = mint.round(input)
>>> print(output)
[ 1.  2.  2.  2. -4.]
>>> input = Tensor(np.array([0.81, 1.52, 2.35, 2.53, -4.57]), mindspore.float32)
>>> output = mint.round(input, decimals=1)
>>> print(output)
[ 0.8  1.5  2.4  2.5 -4.6]