concat api support empty tensor. (#35845)

PaddlePaddle · Sep 24, 2021 · eb28a36 · eb28a36
1 parent 0c0817c
commit eb28a36
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/paddle/fluid/layers/tensor.py b/python/paddle/fluid/layers/tensor.py
@@ -322,6 +322,8 @@ def concat(input, axis=0, name=None):
         if isinstance(axis, Variable):
             axis = axis.numpy()
             axis = axis.item(0)
+        if not isinstance(input, Variable):
+            input = [t for t in input if t.shape.count(0) == 0]
         return _C_ops.concat(input, 'axis', axis)
 
     check_type(input, 'input', (list, tuple, Variable), 'concat')