diff --git a/python/paddle/fluid/layers/tensor.py b/python/paddle/fluid/layers/tensor.py index 3a8dfdc858079c34d749e07158a61b4add553096..5fa110e4db0fa7e190c4cd65e73912bd08c509fe 100644 --- a/python/paddle/fluid/layers/tensor.py +++ b/python/paddle/fluid/layers/tensor.py @@ -678,8 +678,7 @@ def assign(input, output=None): raise ValueError("The size of input is too big. Please consider " "saving it to file and 'load_op' to load it") if output is None: - output = helper.create_variable_for_type_inference( - dtype=input.dtype) + output = helper.create_variable_for_type_inference(dtype=dtype) helper.append_op( type='assign_value', outputs={'Out': [output]},