[object Object][object Object]

[object Object]

获取每行最大值对应列索引，或同时获取每行最大值及其对应列索引。

[object Object]

设 [object Object]，[object Object]。对 [object Object]：

$$\mathrm{dst}_{i,0} = \underset{0 \le j < C}{\operatorname{argmax}} \; \mathrm{src}_{i,j}$$

$$\mathrm{dstval}_{i,0} = \max_{0 \le j < C} \mathrm{src}_{i,j}$$

[object Object]

同步形式：

Lowering may introduce internal scratch tiles; the C++ intrinsic requires an explicit [object Object] operand.

[object Object][object Object][object Object]

声明于 [object Object]:

[object Object]

仅输出索引：

同时输出值和索引：

[object Object][object Object]

• 支持的源元素类型：[object Object]、[object Object]。
• [object Object] 必须使用标准ND布局：行主且非分形（[object Object]、[object Object]）。
• 仅输出索引时：
  • [object Object] 和 [object Object] 必须为 [object Object]。
  • 支持的目标元素类型：[object Object]、[object Object]。
  • 运行时检查遵循共享的行归约检查路径：
    • [object Object]
    • [object Object]
    • [object Object]
  • [object Object] 通过共享的行归约索引检查路径约束，可使用以下任一非分形布局：
    • 单列DN布局（[object Object]、[object Object]），或
    • 有效列数为1的ND布局。
• 同时输出值和索引时：
  • [object Object]、[object Object]、[object Object] 必须为 [object Object]。
  • [object Object]的元素类型必须与[object Object]的元素类型一致。
  • 支持的目标元素类型：
    • 源元素类型为[object Object]时，支持[object Object]、[object Object]。
    • 源元素类型为[object Object]时，支持[object Object]、[object Object]。
  • 运行时检查遵循共享的行归约检查路径：
    • [object Object]
    • [object Object]
    • [object Object]
    • [object Object]
  • [object Object]、[object Object]通过共享的行归约索引检查路径约束，可使用以下任一非分形布局：
    • 单列DN布局（[object Object]、[object Object]），或
    • 有效列数为1的ND布局。

[object Object]

• 仅Atlas A2/A3 训练系列产品/Atlas A2/A3 推理系列产品使用[object Object]临时Tile，Ascend 950PR/Ascend 950DT接收[object Object]但实际并不使用。
• Atlas A2/A3 训练系列产品/Atlas A2/A3 推理系列产品实现根据 [object Object] 与 [object Object]（以下缩写为 [object Object]）的关系选择三条代码路径之一：

[object Object]

• 仅输出索引模式：[object Object] 不使用。硬件 [object Object] 指令直接写入 [object Object]。
• 值+索引模式：[object Object] 用作小型缓冲区（每行2个元素：一个值 + 一个索引）。[object Object] 可使用以下任一非分形布局：
  • 单列DN布局（[object Object]、[object Object]），有效行数为 [object Object]。
  • 有效行数为 [object Object] 且有效列数为2的ND布局。

[object Object]

• [object Object] 被使用于单阶段归约。
• [object Object] tile的行数与 [object Object] 相同。
• [object Object] tile每行所需stride按以下公式计算：

[object Object]

• [object Object] 被使用于两阶段归约，所需空间大于单阶段。
• [object Object] tile的行数与 [object Object] 相同。
• [object Object] tile每行所需stride按以下公式计算：

• [object Object] 用于存放每行tmp区域末尾的最终值+索引结果。

[object Object][object Object][object Object][object Object][object Object][object Object][object Object]