Round

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品
Atlas 200I/500 A2 推理产品 ×
Atlas 推理系列产品 ×
Atlas 训练系列产品

功能说明

  • 将输入张量的值舍入到最接近的整数,若该值与两个整数距离一样则向偶数取整。
  • 将输入张量的元素四舍五入到指定的位数。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 输入张量 BFLOAT16、FLOAT16、 FLOAT32、 DOUBLE、 INT32、 INT64 ND
decimals 属性 一个可选的整数属性,指定要保留的小数位数。默认值为0 INT -
y 输出 输出张量 同输入张量x ND

约束说明

  • 当输入值在[-0.5, -0]之间时,输出值为0。
  • 针对decimals不为0的场景:输入数据超过(-347000, 347000)范围,精度可能会有影响。

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_round 通过aclnnRound接口方式调用Round算子
图模式调用 test_geir_round 通过算子IR构图方式调用Round算子