mathematical-optimization - Torch の LogSoftMax の入力に対する勾配 wrt を計算するコードを理解しようとしています

Question

コードの出典: https://github.com/torch/nn/blob/master/lib/THNN/generic/LogSoftMax.c

このコードがモジュール LogSoftMax の入力に対する勾配 wrt をどのように計算しているかわかりません。私が混乱しているのは、2 つの for ループが何をしているのかです。

for (t = 0; t < nframe; t++)
{
sum = 0;
gradInput_data = gradInput_data0 + dim*t;
output_data = output_data0 + dim*t;
gradOutput_data = gradOutput_data0 + dim*t;

for (d = 0; d < dim; d++)
  sum += gradOutput_data[d];

for (d = 0; d < dim; d++)
  gradInput_data[d] = gradOutput_data[d] - exp(output_data[d])*sum;
 }
}