se_atten_grad

deepmodeling · Oct 9, 2023 · 39bc5c7 · 39bc5c7
1 parent 7667a4b
commit 39bc5c7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/op/_tabulate_grad.py b/deepmd/op/_tabulate_grad.py
@@ -55,7 +55,7 @@ def _tabulate_fusion_se_atten_grad_cc(op, dy):
         op.outputs[0],
         is_sorted=op.get_attr("is_sorted"),
     )
-    return [None, None, dy_dx, dy_df, None]
+    return [None, None, dy_dx, dy_df, dy_dtwo]
 
 
 @ops.RegisterGradient("TabulateFusionSeAttenGrad")