limit concat_op

sljlp · Dec 2, 2022 · e40c51f · e40c51f
1 parent 2a81756
commit e40c51f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/paddle/tensor/manipulation.py b/python/paddle/tensor/manipulation.py
@@ -1127,6 +1127,10 @@ def concat(x, axis=0, name=None):
             #  [14 15 16]]
     """
     input = x
+
+    if not isinstance(input, Variable) and input[0].shape == []:
+        raise ValueError("zero-dimensional arrays cannot be concatenated")
+
     if in_dygraph_mode():
         if isinstance(axis, Variable):
             axis = axis.numpy()