fix bug of grad

PaddlePaddle · May 30, 2023 · 0444f78 · 0444f78
1 parent b8a141f
commit 0444f78
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/python/paddle/fluid/tests/unittests/test_lerp_op.py b/python/paddle/fluid/tests/unittests/test_lerp_op.py
@@ -255,8 +255,16 @@ def init_wshape(self):
         self.wshape = [1]
 
     def init_grad(self, w):
-        self.x_grad = np.ones(self.xshape) * (1 - w)
-        self.y_grad = np.ones(self.xshape) * w
+        self.x_grad = (
+            np.ones(self.xshape)
+            * (1 - w)
+            / (np.prod(self.xshape) / np.prod(self.wshape))
+        )
+        self.y_grad = (
+            np.ones(self.yshape)
+            * w
+            / (np.prod(self.yshape) / np.prod(self.wshape))
+        )
 
     def test_check_output(self):
         place = core.CUDAPlace(0)