CANN ops-nn GEMMV3矩阵乘法算子

GemmV3【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn产品支持情况产品是否支持昇腾910_95 AI处理器√Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√Atlas 200I/500 A2推理产品×Atlas 推理系列产品×Atlas 训练系列产品×功能说明算子功能计算α乘以A与B的乘积再与β和input C的乘积求和。计算公式$$ outα(A B) βC $$ 其中$op(A)$$op(B)$ 和 $op(C)$ 分别是维度为 $(M, K)$, $(K, N)$ 和 $(M, N)$的矩阵。$α$$β$是标量常数。参数说明参数名输入/输出/属性描述数据类型数据格式a输入矩阵乘运算中的左矩阵。FLOAT16, BFLOAT16NDb输入矩阵乘运算中的右矩阵。FLOAT16, BFLOAT16NDalpha输入与a、b矩阵乘结果相乘的标量常数。FLOAT16, BFLOAT16NDbeta输入与input c相乘的标量常数。FLOAT16, BFLOAT16NDc输出输入input和输出进行原地累加。FLOAT32ND约束说明不支持空tensor。支持连续tensor非连续tensor只支持转置场景。调用说明调用方式样例代码说明aclnn接口test_aclnn_gemmv3参考示例算子调用aclnnGemmv3接口方式调用Gemmv3算子【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考