BroadcastTo算子样例
概述
样例概述：本样例介绍了利用ATVC实现BroadcastTo单算子并完成功能验证
- 算子功能：对输入tensor的指定轴进行广播计算并输出结果
- 使用的ATVC模板：Broadcast
- 调用方式：Kernel直调
样例支持AI处理器型号：
- Ascend 910C
- Ascend 910B
算子描述
BroadcastTo是对输入tensor的指定轴进行广播计算并输出结果的Broadcast类算子。
BroadcastTo算子规格：
| 算子类型(OpType) | BroadcastTo | |||
| 算子输入 | ||||
| name | shape | data type | format | |
| x | 1 * 2048 | float | ND | |
| 算子输出 | ||||
| y | 8 * 2048 | float | ND | |
| 核函数名 | BroadcastCustom | |||
目录结构
| 文件名 | 描述 |
|---|---|
| broadcast_to.cpp | BroadcastTo算子代码实现以及调用样例 |
算子运行
在ascendc-api-adv代码仓目录下执行：
cd ./ops_templates/atvc/examples
bash run_examples.sh broadcast_to