fix the bug

Author: bapijun

python/paddle/tensor/__init__.py

@@ -328,6 +328,8 @@
 from .math import i1e  # noqa: F401
 from .math import polygamma  # noqa: F401
 from .math import polygamma_  # noqa: F401
+from .math import renorm  # noqa: F401
+from .math import renorm_  # noqa: F401
 
 from .random import multinomial  # noqa: F401
 from .random import standard_normal  # noqa: F401
@@ -376,6 +378,9 @@
 
 from .einsum import einsum  # noqa: F401
 
+from ..signal import istft  # noqa: F401
+from ..signal import stft  # noqa: F401
+
 # this list used in math_op_patch.py for _binary_creator_
 tensor_method_func = [  # noqa
     'create_parameter',
@@ -409,6 +414,7 @@
     'all',
     'any',
     'asin',
+    'asin_',
     'atan',
     'ceil',
     'ceil_',
@@ -492,8 +498,6 @@
     'add_',
     'subtract',
     'subtract_',
-    'atan',
-    'logsumexp',
     'inverse',
     'log1p',
     'log1p_',
@@ -632,6 +636,7 @@
     'triangular_solve',
     'asinh',
     'atanh',
+    'atanh_',
     'acosh',
     'lu',
     'lu_unpack',
@@ -687,6 +692,61 @@
     'i1e',
     'polygamma',
     'polygamma_',
+    'atan2',
+    'diagflat',
+    'multinomial',
+    'pinv',
+    'renorm',
+    'renorm_',
+    'tan',
+    'tan_',
+    'tril',
+    'tril_',
+    'triu',
+    'triu_',
+    'stft',
+    'istft',
+    'abs_',
+    'acos_',
+    'atan_',
+    'cos_',
+    'cosh_',
+    'sin_',
+    'sinh_',
+    'acosh_',
+    'asinh_',
+    'diag',
+    'eye',
+    'linspace',
+    'fill_constant',
+    'ones',
+    'ones_like',
+    'zeros',
+    'zeros_like',
+    'arange',
+    'full',
+    'full_like',
+    'meshgrid',
+    'empty',
+    'empty_like',
+    'complex',
+    'eigh',
+    'standard_normal',
+    'normal',
+    'uniform',
+    'randn',
+    'rand',
+    'randint',
+    'randint_like',
+    'randperm',
+    'poisson',
+    'searchsorted',
+    'set_printoptions',
+    'array_length',
+    'array_read',
+    'array_write',
+    'create_array',
+    'einsum',
     'normal_',
 ]
 

python/paddle/tensor/manipulation.py

@@ -1984,8 +1984,10 @@ def split(x, num_or_sections, axis=0, name=None):
             dim = (len(input.shape) + dim) if dim < 0 else dim
 
         if isinstance(num_or_sections, int):
+            dim = dim if dim >= 0 else dim + len(input.shape)
             return _C_ops.split_with_num(input, num_or_sections, dim)
         else:
+            dim = dim if dim >= 0 else dim + len(input.shape)
             return _C_ops.split(input, num_or_sections, dim)
 
     else:
@@ -3168,7 +3170,7 @@ def tile(x, repeat_times, name=None):
             # Tensor(shape=[1, 6], dtype=int32, place=Place(gpu:0), stop_gradient=True,
             #        [[1, 2, 3, 1, 2, 3]])
     """
-    if in_dynamic_mode():
+    if in_dynamic_or_pir_mode():
         if isinstance(repeat_times, core.eager.Tensor):
             assert (
                 repeat_times.ndim == 1