| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
个人-AscendC实现ReduceMean算子贡献 Co-authored-by: xiaopei-1<1550747408@qq.com> # message auto-generated for no-merge-commit merge: !393 merge reducemean into master 个人-AscendC实现ReduceMean算子贡献 Created-by: xiaopei-1 Commit-by: xiaopei-1 Merged-by: cann-robot Description: ## 描述 使用AscendC对TBE实现的ReduceMean算子进行重构，实现了AscendC实现的算子对Atlas 200/500 A2推理产品和Atlas 800I/T A2硬件的适配。 ## 关联的Issue 关联Issue：https://gitcode.com/cann/ops-math/issues/240 ## 测试 测试用例  测试结果  ## 文档更新 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!393 | 5 个月前 | |
个人-AscendC实现ReduceMean算子贡献 Co-authored-by: xiaopei-1<1550747408@qq.com> # message auto-generated for no-merge-commit merge: !393 merge reducemean into master 个人-AscendC实现ReduceMean算子贡献 Created-by: xiaopei-1 Commit-by: xiaopei-1 Merged-by: cann-robot Description: ## 描述 使用AscendC对TBE实现的ReduceMean算子进行重构，实现了AscendC实现的算子对Atlas 200/500 A2推理产品和Atlas 800I/T A2硬件的适配。 ## 关联的Issue 关联Issue：https://gitcode.com/cann/ops-math/issues/240 ## 测试 测试用例  测试结果  ## 文档更新 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!393 | 5 个月前 | |
个人-AscendC实现ReduceMean算子贡献 Co-authored-by: xiaopei-1<1550747408@qq.com> # message auto-generated for no-merge-commit merge: !393 merge reducemean into master 个人-AscendC实现ReduceMean算子贡献 Created-by: xiaopei-1 Commit-by: xiaopei-1 Merged-by: cann-robot Description: ## 描述 使用AscendC对TBE实现的ReduceMean算子进行重构，实现了AscendC实现的算子对Atlas 200/500 A2推理产品和Atlas 800I/T A2硬件的适配。 ## 关联的Issue 关联Issue：https://gitcode.com/cann/ops-math/issues/240 ## 测试 测试用例  测试结果  ## 文档更新 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!393 | 5 个月前 | |
个人-AscendC实现ReduceMean算子贡献 Co-authored-by: xiaopei-1<1550747408@qq.com> # message auto-generated for no-merge-commit merge: !393 merge reducemean into master 个人-AscendC实现ReduceMean算子贡献 Created-by: xiaopei-1 Commit-by: xiaopei-1 Merged-by: cann-robot Description: ## 描述 使用AscendC对TBE实现的ReduceMean算子进行重构，实现了AscendC实现的算子对Atlas 200/500 A2推理产品和Atlas 800I/T A2硬件的适配。 ## 关联的Issue 关联Issue：https://gitcode.com/cann/ops-math/issues/240 ## 测试 测试用例  测试结果  ## 文档更新 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!393 | 5 个月前 | |
个人-AscendC实现ReduceMean算子贡献 Co-authored-by: xiaopei-1<1550747408@qq.com> # message auto-generated for no-merge-commit merge: !393 merge reducemean into master 个人-AscendC实现ReduceMean算子贡献 Created-by: xiaopei-1 Commit-by: xiaopei-1 Merged-by: cann-robot Description: ## 描述 使用AscendC对TBE实现的ReduceMean算子进行重构，实现了AscendC实现的算子对Atlas 200/500 A2推理产品和Atlas 800I/T A2硬件的适配。 ## 关联的Issue 关联Issue：https://gitcode.com/cann/ops-math/issues/240 ## 测试 测试用例  测试结果  ## 文档更新 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!393 | 5 个月前 | |
fix: 修正 experimental/math/reduce_*_v2/README.md 文档中算子名字与实际不一致问题 Co-authored-by: baidu_41609401<1294651225@qq.com> # message auto-generated for no-merge-commit merge: !2669 merge master into master fix: 修正 experimental/math/reduce_*_v2/README.md 文档中算子名字与实际不一致问题 Created-by: baidu_41609401 Commit-by: baidu_41609401 Merged-by: cann-robot Description: ## 描述 修改文档中算子名字与实际不一致情况，共修改3个文档，详细情况如下： | 文件路径 | 修改前 | 修改后 | |--|--|--| | ops-math/experimental/math/reduce_max_v2/README.md | ReduceMax | ReduceMaxV2 | | ops-math/experimental/math/reduce_mean_v2/README.md | ReduceMean | ReduceMeanV2 | | ops-math/experimental/math/reduce_min_v2/README.md | ReduceMin | ReduceMinV2 | ## 关联的Issue https://gitcode.com/cann/ops-math/issues/1497 https://gitcode.com/cann/ops-math/issues/1501 ## 文档更新 更新了以下三个文档： ops-math/experimental/math/reduce_max_v2/README.md ops-math/experimental/math/reduce_mean_v2/README.md ops-math/experimental/math/reduce_min_v2/README.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!2669 | 3 个月前 |
ReduceMeanV2
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件 | √ |
功能说明
- 算子功能：对输入张量x沿指定维度计算给定张量的求平均。
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| self | 输入 | x | float、half | ND |
| axis | 属性 | axes | int | ND |
| keepdims | 属性 | keepdims | int | ND |
| y | 输出 | y | float、half | ND |
约束说明
无
待更新
目前支持float32、float16数据类型，仅支持2维，axis仅支持1根轴reduce，keepdims支持0/1，后续会持续改进
调用说明
| 调用方式 | 样例代码 | 说明 |
|---|---|---|
| aclnn接口 | test_reduce_mean_v2 | 通过[aclnnReduceMeanV2]接口方式调用reduce_mean算子。 |