Addcmul

【免费下载链接】ops-math 本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。 【免费下载链接】ops-math 项目地址: https://gitcode.com/cann/ops-math

贡献说明

贡献者 贡献方 贡献算子 贡献时间 贡献内容
skywang2 个人开发者 Addcmul 2025/12/31 新增Addcmul算子

支持的产品型号

  • Atlas A2训练系列产品

产品形态详细说明请参见昇腾产品形态说明

算子描述

  • 功能描述

    Addcmul算子实现了向量x1乘向量x2,乘标量value后的结果再加上向量input_data,返回计算结果的功能。

  • 原型信息

    算子类型(OpType) Addcmul
    name Type data type format
    算子输入
    input_data tensor float32,float16,int32,bfloat16 ND
    x1 tensor float32,float16,int32,bfloat16 ND
    x2 tensor float32,float16,int32,bfloat16 ND
    value scalar float32,float16,int32,bfloat16 -
    算子输出 y tensor float32,float16,int32,bfloat16 ND
    核函数名 addcmul

约束与限制

  • input_data, x1, x2, y, value, out的数据类型只支持float32,float16,int32,bfloat16,数据格式只支持ND

运行验证

测试命令调用方式:build.sh

目录 描述
test_aclnn_addcmul.cpp 通过aclnn调用的方式调用Addcmul算子。

【免费下载链接】ops-math 本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。 【免费下载链接】ops-math 项目地址: https://gitcode.com/cann/ops-math

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐