concat api support empty tensor. (#35845) (#36096)

PaddlePaddle · Sep 26, 2021 · bc13ab9 · bc13ab9
1 parent effb70f
commit bc13ab9
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')