avoid calling cast twice (#30527)

4 years ago · 34bf8dfc40
parent bdae7ed326
commit 34bf8dfc40
1 changed files with 1 additions and 0 deletions
--- a/python/paddle/fluid/layers/tensor.py
+++ b/python/paddle/fluid/layers/tensor.py
@ -228,6 +228,7 @@ def cast(x, dtype):
        if not isinstance(dtype, core.VarDesc.VarType):
            dtype = convert_np_dtype_to_dtype_(dtype)
        out = core.ops.cast(x, 'in_dtype', x.dtype, 'out_dtype', dtype)
+        return out

    check_variable_and_dtype(
        x, 'x',