refine creation.py

Author: HydrogenSulfate

python/paddle/tensor/creation.py

@@ -1518,8 +1518,8 @@ def zeros_like(
 
 
 def eye(
-    num_rows: int,
-    num_columns: int | None = None,
+    num_rows: int | paddle.Tensor,
+    num_columns: int | paddle.Tensor | None = None,
     dtype: DTypeLike | None = None,
     name: str | None = None,
     *,
@@ -1532,8 +1532,8 @@ def eye(
     This function constructs 2-D Tensor with ones on the diagonal and zeros elsewhere.
 
     Args:
-        num_rows(int): the number of rows in each batch Tensor.
-        num_columns(int|None, optional): the number of columns in each batch Tensor.
+        num_rows(int | paddle.Tensor): the number of rows in each batch Tensor.
+        num_columns(int | paddle.Tensor | None, optional): the number of columns in each batch Tensor.
             If None, default: num_rows.
         dtype(np.dtype|str, optional): The data type of the returned Tensor.
             It should be int32, int64, float16, float32, float64, complex64, complex128. Default: if None, the data type
@@ -2795,6 +2795,7 @@ def diag(
         return out
 
 
+@SizeArgsDecorator()
 def empty(
     shape: ShapeLike,
     dtype: DTypeLike | None = None,