归约操作算子样例介绍
概述
本样例集介绍了归约操作算子不同特性的典型用法,给出了对应的端到端实现。
算子开发样例
| 目录名称 | 功能描述 |
|---|---|
| mean | 本样例介绍了基于Ascend C异构混合编程模型演示Mean融合算子的核函数直调实现,根据最后一轴的方向对各元素求平均值 |
| reduce | 本样例介绍了调用WholeReduceSum和BlockReduceSum高阶API实现reduce算子,实现了连续内存上数据元素的累加,返回累加结果的功能 |
| reducemax | 本样例介绍了调用ReduceMax高阶API实现reducemax算子,在所有的输入数据中找出最大值及最大值对应的索引位置 |
| reducemean | 本样例演示了基于ReduceMean高阶API实现reducemean算子。样例对一个多维向量按照指定的维度求平均值 |
| reducemin | 本样例演示了调用ReduceMin高阶API实现reducemin算子。样例对一个多维向量在指定的维度求最小值 |
| reducesum | 本样例演示了基于ReduceSum高阶API实现reducesum算子。样例对一个多维向量按照指定的维度进行数据累加 |
| sum | 本样例介绍了调用Sum高阶API实现sum算子,获取输入数据最后一个维度的元素总和 |