mindspore.ops.round
- mindspore.ops.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。
- 返回:
Tensor,shape和数据类型与输入 input 相同。
- 异常:
TypeError - input 不是Tensor。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor, ops >>> input = Tensor(np.array([0.8, 1.5, 2.3, 2.5, -4.5]), mindspore.float32) >>> output = ops.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 = ops.round(input, decimals=1) >>> print(output) [ 0.8 1.5 2.4 2.5 -4.6]