update the paddle backend of ivy.stack to work for empty arrays

sushmanthreddy · Aug 17, 2023 · 3a8c7b6 · 3a8c7b6
1 parent 52a4b32
commit 3a8c7b6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/ivy/functional/backends/paddle/manipulation.py b/ivy/functional/backends/paddle/manipulation.py
@@ -231,6 +231,12 @@ def stack(
 
     arrays = list(map(lambda x: x.cast(dtype), arrays))
 
+    first_shape = arrays[0].shape
+    if not all(arr.shape == first_shape for arr in arrays):
+        raise Exception("Shapes of all inputs must match")
+    if 0 in first_shape:
+        return ivy.empty(first_shape[:axis] + [len(arrays)] + first_shape[axis:], dtype=dtype)
+
     if dtype in [paddle.int8, paddle.int16, paddle.uint8, paddle.float16, paddle.bool]:
         arrays = list(map(lambda x: x.cast("float32"), arrays))
         return paddle.stack(arrays, axis=axis).cast(dtype)