[Argmax]、[Max] - 4.1 日本語

Save Argmax オプションを使用すると、出力を DDR 空間に回復する際に、チャネルの次元に沿った Argmax および Max が有効になります。セグメンテーションのように、最大値のインデックスだけが必要な場合は、モデルの softmax を argmax に置き換えることで、Exp 計算を省略して遅延を削減できます。