Round【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√Atlas 200I/500 A2 推理产品×Atlas 推理系列产品×Atlas 训练系列产品√功能说明将输入张量的值舍入到最接近的整数若该值与两个整数距离一样则向偶数取整。将输入张量的元素四舍五入到指定的位数。参数说明参数名输入/输出/属性描述数据类型数据格式x输入输入张量BFLOAT16、FLOAT16、 FLOAT32、 DOUBLE、 INT32、 INT64NDdecimals属性一个可选的整数属性指定要保留的小数位数。默认值为0INT-y输出输出张量同输入张量xND约束说明当输入值在[-0.5, -0]之间时输出值为0。针对decimals不为0的场景输入数据超过(-347000, 347000)范围精度可能会有影响。调用说明调用方式调用样例说明aclnn调用test_aclnn_round通过aclnnRound接口方式调用Round算子图模式调用test_geir_round通过算子IR构图方式调用Round算子【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考