PaddlePaddle
diff --git a/‎python/paddle/base/layer_helper_base.py
Copy file name to clipboard
+2-1Lines changed: 2 additions & 1 deletion b/‎python/paddle/base/layer_helper_base.py
Copy file name to clipboard
+2-1Lines changed: 2 additions & 1 deletion
diff --git a/‎python/paddle/tensor/attribute.py
Copy file name to clipboard
+17-12Lines changed: 17 additions & 12 deletions b/‎python/paddle/tensor/attribute.py
Copy file name to clipboard
+17-12Lines changed: 17 additions & 12 deletions
@@ -11,6 +11,7 @@
 # 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.
+from __future__ import annotations
 
 import copy
 
@@ -460,7 +461,7 @@ def create_parameter(
 
     def create_variable_for_type_inference(
         self, dtype, stop_gradient=False, shape=None
-    ):
+    ) -> paddle.Tensor:
         """Create a temporary variable that should be type inferred layer.
 
         Note:
 
@@ -12,7 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-# TODO: define functions to get tensor attributes
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
 
 import numpy as np
 
@@ -25,10 +27,13 @@
 from ..framework import LayerHelper, core
 from .creation import _complex_to_real_dtype, assign
 
+if TYPE_CHECKING:
+    from paddle import Tensor
+
 __all__ = []
 
 
-def rank(input):
+def rank(input: Tensor) -> Tensor:
     """
 
     Returns the number of dimensions for a tensor, which is a 0-D int32 Tensor.
@@ -56,7 +61,7 @@ def rank(input):
     return out
 
 
-def shape(input):
+def shape(input: Tensor) -> Tensor:
     """
     Get the shape of the input.
 
@@ -97,14 +102,14 @@ def shape(input):
             >>> exe = paddle.static.Executor(paddle.CPUPlace())
             >>> exe.run(paddle.static.default_startup_program())
 
-            >>> img = np.ones((3, 100, 100)).astype(np.float32)
+            >>> img = np.ones((3, 100, 100)).astype(np.float32) # type: ignore
 
             >>> res = exe.run(paddle.static.default_main_program(), feed={'x':img}, fetch_list=[output])
             >>> print(res)
             [array([  3, 100, 100], dtype=int32)]
     """
     if in_dynamic_or_pir_mode():
-        out = _C_ops.shape(input)
+        out = _C_ops.shape(input)  # type: ignore
         out.stop_gradient = True
         return out
     else:
@@ -139,7 +144,7 @@ def shape(input):
         return out
 
 
-def is_complex(x):
+def is_complex(x: Tensor) -> bool:
     """Return whether x is a tensor of complex data type(complex64 or complex128).
 
     Args:
@@ -179,7 +184,7 @@ def is_complex(x):
     return is_complex_dtype
 
 
-def is_floating_point(x):
+def is_floating_point(x: Tensor) -> bool:
     """
     Returns whether the dtype of `x` is one of paddle.float64, paddle.float32, paddle.float16, and paddle.bfloat16.
 
@@ -219,7 +224,7 @@ def is_floating_point(x):
     return is_fp_dtype
 
 
-def is_integer(x):
+def is_integer(x: Tensor) -> bool:
     """Return whether x is a tensor of integral data type.
 
     Args:
@@ -272,13 +277,13 @@ def is_integer(x):
     return is_int_dtype
 
 
-def real(x, name=None):
+def real(x: Tensor, name: str | None = None) -> Tensor:
     """
     Returns a new Tensor containing real values of the input Tensor.
 
     Args:
         x (Tensor): the input Tensor, its data type could be complex64 or complex128.
-        name (str, optional): The default value is None. Normally there is no need for
+        name (str|None, optional): The default value is None. Normally there is no need for
             user to set this property. For more information, please refer to :ref:`api_guide_Name` .
 
     Returns:
@@ -320,13 +325,13 @@ def real(x, name=None):
         return out
 
 
-def imag(x, name=None):
+def imag(x: Tensor, name: str | None = None) -> Tensor:
     """
     Returns a new tensor containing imaginary values of input tensor.
 
     Args:
         x (Tensor): the input tensor, its data type could be complex64 or complex128.
-        name (str, optional): The default value is None. Normally there is no need for
+        name (str|None, optional): The default value is None. Normally there is no need for
             user to set this property. For more information, please refer to :ref:`api_guide_Name` .
 
     Returns: