【SCU】【Paddle Tensor No.17、20】新增 paddle.inf, paddle.newaxis, 复用math.inf、None (#69464)

Sylence8 · web-flow · commit cabe06d92341 · 2024-11-20T11:55:37.000+08:00
* 【SCU】【Paddle Tensor No.17、20】新增 paddle.inf, paddle.newaxis

* check codestyle

* merge constants test files

* fix codestyle

* Update __init__.py

* Update __init__.py
diff --git a/python/paddle/__init__.py b/python/paddle/__init__.py
@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import math
 import typing
 
 __is_metainfo_generated = False
@@ -771,6 +772,10 @@
 ir_guard = IrGuard()
 ir_guard._switch_to_pir()
 
+# Constants
+newaxis: None = None
+inf = math.inf
+
 __all__ = [
     'block_diag',
     'iinfo',
@@ -1204,4 +1209,6 @@
     'positive',
     'from_dlpack',
     'to_dlpack',
+    'inf',
+    'newaxis',
 ]
diff --git a/test/legacy_test/test_constant.py b/test/legacy_test/test_constant.py
@@ -0,0 +1,31 @@
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import unittest
+
+import numpy as np
+
+import paddle
+
+
+class TestConstant(unittest.TestCase):
+    def test_inf(self):
+        x = np.array([paddle.inf])
+        np.testing.assert_equal(repr(x), 'array([inf])')
+
+    def test_none_index(self):
+        # `None` index adds newaxis
+        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)
