[fix assign op]/2 using assign op instead of scale1.0

7 years ago · 19a28b1611
parent 8806801eb8
commit 19a28b1611
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/fluid/layers/tensor.py
+++ b/python/paddle/fluid/layers/tensor.py
@ -195,7 +195,7 @@ def assign(input, output):
        helper.append_op(
            type='assign',
            inputs={'X': [input]},
-            outputs={'Out': [output]},
+            outputs={'Out': [output]})
    elif isinstance(input, numpy.ndarray):
        dtype = convert_np_dtype_to_dtype_(input.dtype)
        if dtype == VarDesc.VarType.FP32: