diff --git a/python/paddle/fluid/dygraph/layers.py b/python/paddle/fluid/dygraph/layers.py index b56c34b8a3def..e1855ee6db9af 100644 --- a/python/paddle/fluid/dygraph/layers.py +++ b/python/paddle/fluid/dygraph/layers.py @@ -1572,6 +1572,7 @@ def transform(t, device, dtype, blocking): device = t.place if dtype is None: dtype = t.dtype + new_t = t._copy_to(device, blocking) if isinstance(t, framework.ParamBase): if dtype is not None and dtype != t.dtype: