test=allcase (#2139)

PaddlePaddle · Apr 12, 2023 · b82d40e · b82d40e
1 parent ad5bdb6
commit b82d40e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/framework/api/paddlebase/test_prod.py b/framework/api/paddlebase/test_prod.py
@@ -157,7 +157,7 @@ def test_prod10():
     x = np.array([[-0.8, -0.4], [0.7, 0.9]])
     axis = 1
     res = np.prod(x, axis=axis)
-    exp = paddle.prod(paddle.to_tensor(x), axis=paddle.to_tensor(axis))
+    exp = paddle.prod(paddle.to_tensor(x), axis=paddle.to_tensor([axis]))
     assert np.allclose(exp.numpy(), res)