update

PaddlePaddle · Dec 5, 2023 · e5483e0 · e5483e0
1 parent aa4a35e
commit e5483e0
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/rfcs/APIs/20231202_api_design_for_AdaptiveLogSoftmaxWithLoss.md b/rfcs/APIs/20231202_api_design_for_AdaptiveLogSoftmaxWithLoss.md
@@ -27,7 +27,6 @@ $$
 \begin{aligned}
 & \mathrm{ head\_output }=\mathrm{ linear(input, head\_weight, head\_bias) } \\
 & \mathrm{ head\_logprob }=\mathrm{ log\_softmax (head\_output, axis) }=1 \\
-& \mathrm{output} += \mathrm{take\_along\_axis}(\mathrm{head\_logprob}, \mathrm{gather_inds.unsqueeze(1)}, \mathrm{axis}=1).\mathrm{squeeze()} \\
 & \mathrm{loss} = -\mathrm{output.mean()}
 \end{aligned}
 $$