代码拉取完成,页面将自动刷新
67477
Add tensor type axis support for all reduce ops and function
已关闭
What type of PR is this?
/kind task
What does this PR do / why do we need it:
为对外文档中所有Reduce族算子原语及涉及的函数式接口的axis参数添加Tensor类型的支持:
Primitive | function | Tensor |
---|---|---|
ops.ReduceMean | ops.mean | Tensor.mean |
ops.ReduceSum | ops.sum | Tensor.sum |
ops.ReduceProd | ops.prod | Tensor.prod |
ops.ReduceAll | ops.all | Tensor.all |
ops.ReduceAny | ops.any | Tensor.any |
ops.ReduceMax | ops.amax | Tensor.amax |
ops.ReduceMin | ops.amin | Tensor.amin |
ops.ReduceStd | ||
ops.MeanExt | ||
ops.SumExt |
Which issue(s) this PR fixes:
Fixes #
Code review checklist 【代码检视checklist说明】:
-
典型安全编码问题 【违反安全编码案例】
- 是否进行空指针校验
- 是否进行返回值校验 (禁止使用void屏蔽安全函数、自研函数返回值,C++标准库函数确认无问题可以屏蔽)
- 是否正确释放new/malloc申请的内存
-
性能分析 (如果涉及某个子项,请概述设计思想/修改内容)
- 是否修改热点函数 / 算法 / 算子
- 是否考虑并发场景
- 是否考虑通信场景
-
- 是否符合编码规范 【编码规范】
-
- 是否遵守 SOLID原则 / 迪米特法则
-
- 是否涉及模块/特性间交互 (若涉及请概述实现思路)
-
- 是否具备UT测试用例看护 && 测试用例为有效用例 (若新特性无测试用例看护请说明原因)
-
- 是否正确加载、释放秘钥
-
错误处理与记录
- 是否充分考虑接口的异常场景
- 是否正确记录错误信息
Special notes for your reviewers: