Argmax 和 Max - 4.1 简体中文

“Save Argmax”（保存 Argmax）选项会在将输出复原回 DDR 空间时沿通道维度启用 argmax 和 max 特征。在某些情况下（如分割），只需最大值的索引即可。此时将模型中的 softmax 替换为 argmax 即可移除 Exp 计算和缩短时延，这是很有用的。