ReduceMaxV2
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
功能说明
- 算子功能：对输入张量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_max_v2 | 通过[aclnnReduceMaxV2]接口方式调用reduce_max_v2算子。 |