CANN/ops-math偏置加法算子

aclnnBiasAdd【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math产品支持情况产品是否支持Atlas A2 训练系列产品/Atlas A2 推理系列产品√功能说明算子功能为输入张量x的每一个元素加上其对应通道C 维的偏差值bias。计算公式$$out_i x_i bias_{c(i)}$$其中 $c(i)$ 为元素 $i$ 在dataFormat指定的 C通道维上的下标。函数原型每个算子分为两段式接口必须先调用aclnnBiasAddGetWorkspaceSize接口获取入参并根据流程计算所需 workspace 大小再调用aclnnBiasAdd接口执行计算。aclnnStatus aclnnBiasAddGetWorkspaceSize(const aclTensor *x, const aclTensor *bias, char *dataFormatOptional, const aclTensor *out, uint64_t *workspaceSize, aclOpExecutor **executor)aclnnStatus aclnnBiasAdd(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)aclnnBiasAddGetWorkspaceSize参数说明参数名输入/输出描述数据类型数据格式x输入待计算的输入张量FLOAT、FLOAT16、BFLOAT16、INT32NCHW、NHWC、NDHWC、NCDHW、NDbias输入累加偏差长度等于x在dataFormat指定 C 维的大小与 x 一致NDdataFormatOptional输入指定 C 维位置的数据排布可选值 NCHW/NHWC/NDHWC/NCDHW默认 NHWCchar*-out输出计算结果与 x 一致与 x 一致workspaceSize输出返回需要在 Device 侧申请的 workspace 大小uint64_t*-executor输出返回 op 执行器aclOpExecutor**-返回值返回 aclnnStatus 状态码。第一段接口完成入参校验出现以下错误码时请检查返回 161001ACLNN_ERR_PARAM_NULLPTRx、bias、out 任一为空指针。返回 161002ACLNN_ERR_PARAM_INVALIDx、bias、out 的数据类型不一致或不在支持范围bias 长度与 C 维不匹配。aclnnBiasAdd参数说明参数名输入/输出描述数据类型workspace输入在 Device 侧申请的 workspace 内存地址void*workspaceSize输入workspace 大小由第一段接口返回uint64_texecutor输入op 执行器由第一段接口返回aclOpExecutor*stream输入执行任务的 streamaclrtStream返回值返回 aclnnStatus 状态码。约束说明仅支持 Atlas A2 训练系列产品/Atlas A2 推理系列产品 。bias的长度必须等于x在dataFormat指定 C 维上的大小。调用示例完整可运行样例见examples/test_aclnn_bias_add.cpp核心调用流程// 1. 获取 workspace 大小dataFormat 为 char* char dataFormat[] NHWC; uint64_t workspaceSize 0; aclOpExecutor *executor nullptr; aclnnBiasAddGetWorkspaceSize(x, bias, dataFormat, out, workspaceSize, executor); // 2. 申请 workspace若 workspaceSize 0并执行 void *workspaceAddr nullptr; if (workspaceSize 0) { aclrtMalloc(workspaceAddr, workspaceSize, ACL_MEM_MALLOC_HUGE_FIRST); } aclnnBiasAdd(workspaceAddr, workspaceSize, executor, stream); aclrtSynchronizeStream(stream);【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考