fix grad.numpy in test_pylayer_op

PaddlePaddle · Apr 26, 2021 · be50060 · be50060 · paddle-bot-old · Apr 26, 2021
1 parent 4b7242b
commit be50060
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/paddle/fluid/tests/unittests/test_pylayer_op.py b/python/paddle/fluid/tests/unittests/test_pylayer_op.py
@@ -85,7 +85,8 @@ def backward(ctx, dy1, dy2):
         z2 = paddle.tanh(input2) + paddle.tanh(input4)
         z2.mean().backward()
 
-        self.assertTrue(np.max(np.abs((input1.grad - input2.grad))) < 1e-10)
+        self.assertTrue(
+            np.max(np.abs((input1.grad.numpy() - input2.grad.numpy()))) < 1e-10)
 
     def test_simple_pylayer_single_output(self):
         class tanh(PyLayer):