fix reshape shape bug (#58865)

PaddlePaddle · Nov 12, 2023 · 4ff3c4d · 4ff3c4d
1 parent e5f007f
commit 4ff3c4d
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/python/paddle/tensor/manipulation.py b/python/paddle/tensor/manipulation.py
@@ -3865,7 +3865,7 @@ def get_attr_shape(list_shape):
         )
         if isinstance(shape, (list, tuple)):
             if paddle.utils._contain_var(shape):
-                new_shape = paddle.utils._convert_to_tensor_list(shape)
+                new_shape = paddle.utils.get_int_tensor_list(shape)
             else:
                 new_shape = get_attr_shape(shape)
             out = _C_ops.reshape(x, new_shape)

diff --git a/test/legacy_test/test_reshape_op.py b/test/legacy_test/test_reshape_op.py
@@ -732,6 +732,18 @@ def test_static(self):
             self.assertEqual(result[3].shape, (1,))
 
 
+class TestReshapePirOpResultListShape(unittest.TestCase):
+    def test_opresult_list_shape(self):
+        with paddle.pir_utils.IrGuard():
+            x = paddle.static.data(
+                'x',
+                [3],
+            )
+            shape = [1, paddle.full([], 3)]
+            out = paddle.reshape(x, shape)
+            np.testing.assert_array_equal(tuple(out.shape), (-1, -1))
+
+
 if __name__ == "__main__":
     paddle.enable_static()
     unittest.main()