README.md

归约操作算子样例介绍

概述

本样例集介绍了归约操作算子不同特性的典型用法,给出了对应的端到端实现。

样例列表

目录名称 功能描述 支持的产品
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 推理系列产品