【SCU】【Paddle Tensor No.18】新增 paddle.nan, 复用math.nan (#69542)

kineast · web-flow · commit c2feed09c565 · 2024-11-21T22:14:53.000+08:00
* test&amp;init

* test

* test202411201451

* test

* test

* testm

* test

* test

* test

* test
diff --git a/python/paddle/__init__.py b/python/paddle/__init__.py
@@ -772,9 +772,11 @@
 ir_guard = IrGuard()
 ir_guard._switch_to_pir()
 
+
 # Constants
 newaxis: None = None
 inf = math.inf
+nan = math.nan
 
 __all__ = [
     'block_diag',
@@ -1211,4 +1213,5 @@
     'to_dlpack',
     'inf',
     'newaxis',
+    'nan',
 ]
diff --git a/test/legacy_test/test_constant.py b/test/legacy_test/test_constant.py
@@ -29,3 +29,11 @@ def test_none_index(self):
         a = np.array([1, 2, 3])
         np.testing.assert_equal(a[None], a[paddle.newaxis])
         np.testing.assert_equal(a[None].ndim, a.ndim + 1)
+
+    def test_nan(self):
+        x = np.array([paddle.nan])
+        np.testing.assert_equal(repr(x), 'array([nan])')
+
+
+if __name__ == '__main__':
+    unittest.main()
