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