CANN ops-math AdaCast算子

AdaCast【免费下载链接】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 训练系列产品×功能说明算子功能面向 HDRnetHigh Dynamic Range Network实时图像增强网络的输入预处理算子。将传感器输出的 uint16 HDR 图像数据转换为 float16 张量并按白电平pixel完成归一化等价于把「整型→浮点类型转换」与「按除数缩放」两步融合为单次逐元素运算。计算公式$$ out_{i} \mathrm{Cast_FP16}^{satround}\bigl(\mathrm{Cast_FP32}(self_{i}) \times (1 / pixel)\bigr) $$其中末步采用饱和clip 到 fp16 值域 [-65504, 65504] 四舍五入round-to-nearest-even方式转换到 float16。参数说明参数名输入/输出/属性描述数据类型数据格式x输入待归一化的 HDR uint16 图像数据公式中的self_ishape 支持 1D~4D。UINT16NDpixel可选属性白电平归一化基数缩放系数为 1/pixel。默认值为 65535。取值必须为正整数pixel 0。INT64NDy输出归一化后的 float16 张量公式中的out_ishape 与 x 完全一致。FLOAT16ND约束说明数据格式仅支持 ND。输入 x 的 rank 范围为 [1, 4]输出 y 的 shape 必须与 x 完全一致element-wise 语义不支持广播。类型组合固定为 UINT16 → FLOAT16不支持其他 dtype 组合。属性 pixel 必须为正整数pixel 0否则返回 ACLNN_ERR_PARAM_INVALID。支持空 Tensor0 元素直接返回。默认确定性实现相同输入始终产生相同输出。调用说明调用方式调用样例说明图模式调用test_geir_ada_cast通过 GE-IR 构图方式调用 AdaCast 算子。【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考