From aeee5b1e96bc051072153bb26029e90fcf5ab616 Mon Sep 17 00:00:00 2001
From: zxcd <228587199@qq.com>
Date: Fri, 2 Jun 2023 11:30:19 +0800
Subject: [PATCH] Make arange code logic clearer (#54264)

---
 python/paddle/tensor/creation.py | 20 +++++++++++---------
 1 file changed, 11 insertions(+), 9 deletions(-)

diff --git a/python/paddle/tensor/creation.py b/python/paddle/tensor/creation.py
index 7394c247061..d6e378396f3 100644
--- a/python/paddle/tensor/creation.py
+++ b/python/paddle/tensor/creation.py
@@ -1301,16 +1301,18 @@ def arange(start=0, end=None, step=1, dtype=None, name=None):
 
     if dtype is None:
         for val in [start, end, step]:
-            if isinstance(val, Variable) and not val.is_integer():
-                dtype = paddle.get_default_dtype()
-                break
-            elif not isinstance(val, (int, np.integer)) and not isinstance(
-                val, Variable
-            ):
-                dtype = paddle.get_default_dtype()
-                break
+            if isinstance(val, Variable):
+                if not val.is_integer():
+                    dtype = paddle.get_default_dtype()
+                    break
+                else:
+                    dtype = 'int64'
             else:
-                dtype = 'int64'
+                if not isinstance(val, np.integer) and not isinstance(val, int):
+                    dtype = paddle.get_default_dtype()
+                    break
+                else:
+                    dtype = 'int64'
 
     out_shape = None
     if not in_dynamic_mode() and (
-- 
GitLab