22 lines
623 B
ReStructuredText
22 lines
623 B
ReStructuredText
mindspore.ops.round
|
||
====================
|
||
|
||
.. py:function:: mindspore.ops.round(input, *, decimals=0)
|
||
|
||
对输入数据四舍五入到最接近的整数。
|
||
|
||
.. math::
|
||
out_i \approx input_i
|
||
|
||
.. note::
|
||
Ascend平台支持的输入数据类型包括bfloat16（Atlas训练系列产品不支持）、float16、float32、float64、int32、int64。
|
||
|
||
参数：
|
||
- **input** (Tensor) - 输入tensor。
|
||
|
||
关键字参数：
|
||
- **decimals** (int, 可选) - 要舍入到的小数位数。如果为负数，则指定小数点左侧的位数。默认 ``0`` 。
|
||
|
||
返回：
|
||
Tensor
|