归约操作算子样例介绍
概述
本样例集介绍了归约操作算子不同特性的典型用法,给出了对应的端到端实现。
样例列表
| 目录名称 | 功能描述 | 支持的产品 |
|---|---|---|
| reduce | 本样例介绍了调用WholeReduceSum和BlockReduceSum高阶API实现reduce算子,实现了连续内存上数据元素的累加,返回累加结果的功能 | Ascend 950PR/Ascend 950DT Atlas A3 训练系列产品/Atlas A3 推理系列产品 Atlas A2 训练系列产品/Atlas A2 推理系列产品 |
| reducemax | 本样例演示了基于ReduceMax高阶API对一个多维向量按照指定的维度求最大值及最大值对应的索引值。 | Ascend 950PR/Ascend 950DT Atlas A3 训练系列产品/Atlas A3 推理系列产品 Atlas A2 训练系列产品/Atlas A2 推理系列产品 |
| sum | 本样例介绍了调用Sum高阶API实现sum算子,获取输入数据最后一个维度的元素总和 | Ascend 950PR/Ascend 950DT Atlas A3 训练系列产品/Atlas A3 推理系列产品 Atlas A2 训练系列产品/Atlas A2 推理系列产品 |