disable reshape inplace in dygraph model; test=develop (#21157)

5 years ago · 835119c777
parent 5fdfbe3413
commit 835119c777
1 changed files with 2 additions and 2 deletions
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@ -5400,8 +5400,8 @@ def reshape(x, shape, actual_shape=None, act=None, inplace=False, name=None):
                actual_shape.stop_gradient = True
                inputs["Shape"] = actual_shape
-    out = x if inplace else helper.create_variable_for_type_inference(
+    out = x if inplace and not in_dygraph_mode(
-        dtype=x.dtype)
+    ) else helper.create_variable_for_type_inference(dtype=x.dtype)
    x_shape = helper.create_variable_for_type_inference(dtype=x.dtype)
    helper.append_op(
        type="reshape2",