README.md

Abs算子样例

概述

样例概述:本样例介绍了利用ATVOSS实现Abs单算子并完成功能验证

  • Abs算子功能:实现一个Tensor的取绝对值运算

算子描述

  • 算子数学计算公式:

    Abs⁡(in)=∣in∣\operatorname{Abs}(in)= {|in|}

  • 算子参数:

参数名 输入/输出 描述 数据类型 数据格式
in 输入 公式中的Tensor输入。 float ND
out 输出 执行Abs运算后的输出Tensor。 float ND

样例支持的产品

  • Ascend 950PR/Ascend 950DT

目录结构

文件名 描述
abs.cpp Abs样例算子代码实现
CMakeLists.txt Abs样例算子的编译构建文件
README.md Abs样例算子的说明文档

Abs样例算子的编译和运行

  • 编译 在代码仓根目录下执行:
bash scripts/build.sh -DSOC=ascend950 abs
  • 运行 在代码仓目录下执行:
output/bin/abs --help // 查看帮助
output/bin/abs --shape=16,32 // 运行样例