Muls算子样例

概述

样例概述：本样例介绍了利用ATVOSS实现Muls单算子并完成功能验证，重点展示了根据用户不同输入信息，选择不同Compute表达的运算过程

• Muls算子功能：实现一个Tensor和一个Scalar进行乘法运算

算子描述

• 算子数学计算公式：

  $$\mathrm{Muls}(in,scalar)=in\ast scalar$$

• 算子参数：

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

样例支持的产品

• Ascend 950PR/Ascend 950DT

目录结构

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

Muls样例算子的编译和运行

• 编译 在代码仓根目录下执行：

bash scripts/build.sh -DSOC=ascend950 muls

• 运行 在代码仓目录下执行：

output/bin/muls --help // 查看帮助
output/bin/muls --shape=16,32 // 运行样例