From ca27263e8ada1eb4668104850888050a28321e10 Mon Sep 17 00:00:00 2001 From: wangchaochaohu Date: Sat, 5 Sep 2020 14:34:05 +0800 Subject: [PATCH] refine the error of message of linspace test=develop (#27067) (#27070) --- python/paddle/fluid/layers/tensor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/layers/tensor.py b/python/paddle/fluid/layers/tensor.py index 4e247c81314..71f119d4628 100644 --- a/python/paddle/fluid/layers/tensor.py +++ b/python/paddle/fluid/layers/tensor.py @@ -1461,6 +1461,8 @@ def linspace(start, stop, num, dtype=None, name=None): tensor_num = num tensor_start = start tensor_stop = stop + if not isinstance(num, Variable): + check_type(num, 'num', (int), 'linspace') if not isinstance(dtype, core.VarDesc.VarType): dtype = convert_np_dtype_to_dtype_(dtype) if not isinstance(start, Variable): @@ -1486,8 +1488,6 @@ def linspace(start, stop, num, dtype=None, name=None): check_type(stop, 'stop', (int, float), 'linspace') if isinstance(num, Variable): check_dtype(num.dtype, 'num', ['int32'], 'linspace') - else: - check_type(num, 'num', (int), 'linspace') check_dtype(dtype, 'dtype', ['int32', 'int64', 'float32', 'float64'], 'linspace') -- GitLab