CANN/ops-math 偏置加法算子

发布时间:2026/7/10 20:25:03
CANN/ops-math 偏置加法算子 BiasAdd【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math产品支持情况产品是否支持Atlas A2 训练系列产品/Atlas A2 推理系列产品√功能说明算子功能为输入张量的每一个元素加上对应通道channel的偏差值。计算公式$$out_i x_i bias_{c(i)}$$其中 $c(i)$ 为元素 $i$ 在data_format指定的 C通道维上的下标。参数说明参数名输入/输出/属性描述数据类型数据格式x输入待计算的输入张量FLOAT、FLOAT16、BFLOAT16、INT32NCHW、NHWC、NDHWC、NCDHW、NDbias输入累加偏差长度等于 x 的 C 维大小同 xNDy输出计算结果同 x同 xdata_format属性可选指定 C 维位置的数据排布默认 NHWCstring-约束说明仅支持 Atlas A2 训练系列产品/Atlas A2 推理系列产品 。bias的长度必须等于x在data_format指定的 C 维上的大小。实现说明按输入 shape 的结构属性在 host 侧运行时分派多条调度路径schModekernel 侧据此选择对应实现TINY / TINY_NOQUEUE小 total受 kernel launch / prologue 固定开销限制走最薄路径。THIN_TINY_VECTOR_BROADCAST小非对齐 NHWC逐元素标量广播改为向量Copy GatherMaskUB 广播。BROADCAST_UB_TILE整 tile 物化 bias 后做整宽向量Add大 C 非对齐cBlocks 较大经向量广播处理避免标量串行化。bfloat16 作为 dtype decorator额外套Castbf16↔fp32链。调用说明调用方式说明aclnn 单算子调用见examples/test_aclnn_bias_add.cppAPI 由算子定义自动生成aclnnBiasAdd。贡献说明贡献者贡献方贡献算子贡献时间贡献内容杨镇泽gcw_5x5Ew5Ms重庆邮电大学CQUPTBiasAdd2026/06/16新增 BiasAdd 算子Ascend 910B AscendC 实现【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考