修复atleast函数中，”输入为tensor的list，输出不是tensor的list“的bug (#73102)

Qin-sx · web-flow · commit b6b5d60872ad · 2025-06-23T11:31:58.000+08:00
* modified for input of list of tensor in atleast func

	modified:   python/paddle/tensor/manipulation.py
	modified:   test/legacy_test/test_atleast_xd.py

* modified tuple and any in condition

	modified:   python/paddle/tensor/manipulation.py

* added a test for tuple of tensor

	modified:   test/legacy_test/test_atleast_xd.py

* added tests of nested tuple and list

	modified:   test/legacy_test/test_atleast_xd.py
diff --git a/python/paddle/tensor/manipulation.py b/python/paddle/tensor/manipulation.py
@@ -5273,6 +5273,20 @@ def atleast_1d(*inputs, name=None):
             [123]), Tensor(shape=[1, 1], dtype=float32, place=Place(cpu), stop_gradient=True,
             [[1.23000002]])]
     """
+    if len(inputs) == 1 and isinstance(inputs[0], (list, tuple)):
+        if any(
+            isinstance(
+                item,
+                (
+                    paddle.Tensor,
+                    paddle.base.framework.Variable,
+                    paddle.base.libpaddle.pir.Value,
+                ),
+            )
+            for item in inputs[0]
+        ):
+            inputs = inputs[0]
+
     out = []
     for input in inputs:
         if not isinstance(
@@ -5362,6 +5376,20 @@ def atleast_2d(*inputs, name=None):
             [[123]]), Tensor(shape=[1, 1, 1], dtype=float32, place=Place(cpu), stop_gradient=True,
             [[[1.23000002]]])]
     """
+    if len(inputs) == 1 and isinstance(inputs[0], (list, tuple)):
+        if any(
+            isinstance(
+                item,
+                (
+                    paddle.Tensor,
+                    paddle.base.framework.Variable,
+                    paddle.base.libpaddle.pir.Value,
+                ),
+            )
+            for item in inputs[0]
+        ):
+            inputs = inputs[0]
+
     out = []
     for input in inputs:
         if not isinstance(
@@ -5440,6 +5468,20 @@ def atleast_3d(*inputs, name=None):
             [[[123]]]), Tensor(shape=[1, 1, 1, 1], dtype=float32, place=Place(cpu), stop_gradient=True,
             [[[[1.23000002]]]])]
     """
+    if len(inputs) == 1 and isinstance(inputs[0], (list, tuple)):
+        if any(
+            isinstance(
+                item,
+                (
+                    paddle.Tensor,
+                    paddle.base.framework.Variable,
+                    paddle.base.libpaddle.pir.Value,
+                ),
+            )
+            for item in inputs[0]
+        ):
+            inputs = inputs[0]
+
     out = []
     for input in inputs:
         if not isinstance(
diff --git a/test/legacy_test/test_atleast_xd.py b/test/legacy_test/test_atleast_xd.py
@@ -495,5 +495,162 @@ def test_as_tensor_method(self):
                 np.testing.assert_allclose(n, p.numpy(), rtol=RTOL, atol=ATOL)
 
 
+class TestAtleastWithTensorList(unittest.TestCase):
+    """Test when input is a list of paddle tensors"""
+
+    def test_tensor_list_input(self):
+        for device, place in PLACES:
+            paddle.disable_static(place)
+            paddle.set_device(device)
+
+            tensor_list = [
+                paddle.to_tensor(123, dtype='int32'),  # 0D
+                paddle.to_tensor([1, 2, 3], dtype='float32'),  # 1D
+                paddle.to_tensor([[1, 2], [3, 4]], dtype='int64'),  # 2D
+            ]
+
+            # atleast_1d
+            out_1d = paddle.atleast_1d(tensor_list)
+            self.assertTrue(isinstance(out_1d, list))
+            self.assertEqual(len(out_1d), len(tensor_list))
+
+            self.assertEqual(out_1d[0].shape, [1])
+            self.assertEqual(out_1d[1].shape, [3])
+            self.assertEqual(out_1d[2].shape, [2, 2])
+
+            # atleast_2d
+            out_2d = paddle.atleast_2d(tensor_list)
+            self.assertTrue(isinstance(out_2d, list))
+            self.assertEqual(len(out_2d), len(tensor_list))
+
+            self.assertEqual(out_2d[0].shape, [1, 1])
+            self.assertEqual(out_2d[1].shape, [1, 3])
+            self.assertEqual(out_2d[2].shape, [2, 2])
+
+            # atleast_3d
+            out_3d = paddle.atleast_3d(tensor_list)
+            self.assertTrue(isinstance(out_3d, list))
+            self.assertEqual(len(out_3d), len(tensor_list))
+
+            self.assertEqual(out_3d[0].shape, [1, 1, 1])
+            self.assertEqual(out_3d[1].shape, [1, 3, 1])
+            self.assertEqual(out_3d[2].shape, [2, 2, 1])
+
+            np.testing.assert_allclose(
+                out_1d[0].numpy(), [123], rtol=RTOL, atol=ATOL
+            )
+            np.testing.assert_allclose(
+                out_1d[1].numpy(), [1, 2, 3], rtol=RTOL, atol=ATOL
+            )
+            np.testing.assert_allclose(
+                out_1d[2].numpy(), [[1, 2], [3, 4]], rtol=RTOL, atol=ATOL
+            )
+
+
+class TestAtleastWithTensorTuple(unittest.TestCase):
+    """Test when input is a tuple of paddle tensors"""
+
+    def test_tensor_tuple_input(self):
+        for device, place in PLACES:
+            paddle.disable_static(place)
+            paddle.set_device(device)
+
+            tensor_tuple = (
+                paddle.to_tensor(123, dtype='int32'),  # 0D
+                paddle.to_tensor([1, 2, 3], dtype='float32'),  # 1D
+                paddle.to_tensor([[1, 2], [3, 4]], dtype='int64'),  # 2D
+            )
+
+            # atleast_1d
+            out_1d = paddle.atleast_1d(tensor_tuple)
+            self.assertTrue(isinstance(out_1d, list))
+            self.assertEqual(len(out_1d), len(tensor_tuple))
+
+            self.assertEqual(out_1d[0].shape, [1])
+            self.assertEqual(out_1d[1].shape, [3])
+            self.assertEqual(out_1d[2].shape, [2, 2])
+
+            # atleast_2d
+            out_2d = paddle.atleast_2d(tensor_tuple)
+            self.assertTrue(isinstance(out_2d, list))
+            self.assertEqual(len(out_2d), len(tensor_tuple))
+
+            self.assertEqual(out_2d[0].shape, [1, 1])
+            self.assertEqual(out_2d[1].shape, [1, 3])
+            self.assertEqual(out_2d[2].shape, [2, 2])
+
+            # atleast_3d
+            out_3d = paddle.atleast_3d(tensor_tuple)
+            self.assertTrue(isinstance(out_3d, list))
+            self.assertEqual(len(out_3d), len(tensor_tuple))
+
+            self.assertEqual(out_3d[0].shape, [1, 1, 1])
+            self.assertEqual(out_3d[1].shape, [1, 3, 1])
+            self.assertEqual(out_3d[2].shape, [2, 2, 1])
+
+            # Verify values are preserved correctly
+            np.testing.assert_allclose(
+                out_1d[0].numpy(), [123], rtol=RTOL, atol=ATOL
+            )
+            np.testing.assert_allclose(
+                out_1d[1].numpy(), [1, 2, 3], rtol=RTOL, atol=ATOL
+            )
+            np.testing.assert_allclose(
+                out_1d[2].numpy(), [[1, 2], [3, 4]], rtol=RTOL, atol=ATOL
+            )
+
+
+class TestAtleastWithNestedList(unittest.TestCase):
+    """Test when input is a list containing nested lists"""
+
+    def test_nested_list_input(self):
+        for device, place in PLACES:
+            paddle.disable_static(place)
+            paddle.set_device(device)
+
+            nested_list = [[1, 2, 3], [1, 2, 3], [1, 2, 3]]
+
+            # atleast_1d
+            out_1d = paddle.atleast_1d(nested_list)
+            self.assertEqual(out_1d.shape, [3, 3])
+            self.assertTrue(isinstance(out_1d, paddle.Tensor))
+
+            # atleast_2d
+            out_2d = paddle.atleast_2d(nested_list)
+            self.assertEqual(out_2d.shape, [3, 3])
+            self.assertTrue(isinstance(out_2d, paddle.Tensor))
+
+            # atleast_3d
+            out_3d = paddle.atleast_3d(nested_list)
+            self.assertEqual(out_3d.shape, [3, 3, 1])
+            self.assertTrue(isinstance(out_3d, paddle.Tensor))
+
+
+class TestAtleastWithNestedTuple(unittest.TestCase):
+    """Test when input is a tuple containing nested tuples"""
+
+    def test_nested_tuple_input(self):
+        for device, place in PLACES:
+            paddle.disable_static(place)
+            paddle.set_device(device)
+
+            nested_tuple = ((1, 2, 3), (1, 2, 3), (1, 2, 3))
+
+            # atleast_1d
+            out_1d = paddle.atleast_1d(nested_tuple)
+            self.assertEqual(out_1d.shape, [3, 3])
+            self.assertTrue(isinstance(out_1d, paddle.Tensor))
+
+            # atleast_2d
+            out_2d = paddle.atleast_2d(nested_tuple)
+            self.assertEqual(out_2d.shape, [3, 3])
+            self.assertTrue(isinstance(out_2d, paddle.Tensor))
+
+            # atleast_3d
+            out_3d = paddle.atleast_3d(nested_tuple)
+            self.assertEqual(out_3d.shape, [3, 3, 1])
+            self.assertTrue(isinstance(out_3d, paddle.Tensor))
+
+
 if __name__ == '__main__':
     unittest.main()
