Round

Round

产品支持情况

产品 是否支持
Atlas A2 训练系列产品/Atlas A2 推理系列产品

功能说明

  • 算子功能:对输入张量的每一个元素,按照指定的小数位,四舍五入。

  • 计算公式:

    outi=round(inputi,decimals)out_i=round(input_i,decimals)

    举例如下:

    Round(3.56,0) = 4.0

    Round(3.56,1) = 3.5

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 入参,公式中的input,对该数据进行四舍五入 bfloat16,float16,float,int32 ND
decimals 属性 入参,公式中的decimals,指定四舍五入的位数 int ND
y 输出 待进行round计算的出参,公式中的y bfloat16,float16,float,int32 ND

约束说明

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_round.cpp 通过aclnnRound接口方式调用Round算子。

贡献说明

贡献者 贡献方 贡献算子 贡献时间 贡献内容
番茄土豆 南京亚信软件有限公司 Round 2025/12/11 Round算子适配开源仓