Muls算子样例【免费下载链接】atvossATVOSSAscend C Templates for Vector Operator Subroutines是一套基于Ascend C开发的Vector算子库致力于为昇腾硬件上的Vector类融合算子提供极简、高效、高性能、高拓展的编程方式。项目地址: https://gitcode.com/cann/atvoss概述样例概述本样例介绍了利用ATVOSS实现Muls单算子并完成功能验证重点展示了根据用户不同输入信息选择不同Compute表达的运算过程Muls算子功能实现一个Tensor和一个Scalar进行乘法运算算子描述算子数学计算公式 $$ \operatorname{Muls}(in, scalar) {in * scalar} $$算子参数参数名输入/输出描述数据类型数据格式in输入公式中的Tensor输入。float、int32_tNDscalar输入公式中的Scalar输入。floatNDout输出执行Muls运算后的输出Tensor。floatND样例支持的产品Ascend 950PR/Ascend 950DT目录结构文件名描述muls.cppMuls样例算子代码实现CMakeLists.txtMuls样例算子的编译构建文件README.mdMuls样例算子的说明文档Muls样例算子的编译和运行编译 在代码仓根目录下执行bash scripts/build.sh -DSOCascend950 muls运行 在代码仓目录下执行output/bin/muls --help // 查看帮助 output/bin/muls --shape16,32 // 运行样例【免费下载链接】atvossATVOSSAscend C Templates for Vector Operator Subroutines是一套基于Ascend C开发的Vector算子库致力于为昇腾硬件上的Vector类融合算子提供极简、高效、高性能、高拓展的编程方式。项目地址: https://gitcode.com/cann/atvoss创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考