CANN/ATVC BroadcastTo算子样例

【免费下载链接】atvcATVCAscend C Templates for Vector Compute是为基于Ascend C开发的典型Vector算子封装的一系列模板头文件的集合可帮助用户快速开发典型Vector算子。项目地址: https://gitcode.com/cann/atvcBroadcastTo算子样例概述样例概述本样例介绍了利用ATVC实现BroadcastTo单算子并完成功能验证算子功能对输入tensor的指定轴进行广播计算并输出结果使用的ATVC模板Broadcast调用方式Kernel直调样例支持AI处理器型号Ascend 910CAscend 910B算子描述BroadcastTo是对输入tensor的指定轴进行广播计算并输出结果的Broadcast类算子。BroadcastTo算子规格算子类型(OpType)BroadcastTo算子输入nameshapedata typeformatx1 * 2048floatND算子输出y8 * 2048floatND核函数名BroadcastCustom目录结构文件名描述broadcast_to.cppBroadcastTo算子代码实现以及调用样例算子运行在ascendc-api-adv代码仓目录下执行cd ./ops_templates/atvc/examples bash run_examples.sh broadcast_to【免费下载链接】atvcATVCAscend C Templates for Vector Compute是为基于Ascend C开发的典型Vector算子封装的一系列模板头文件的集合可帮助用户快速开发典型Vector算子。项目地址: https://gitcode.com/cann/atvc创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考