fix dygraph gather api

4 years ago · a98c69b6c6
parent e1e666a05f
commit a98c69b6c6
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@ -8311,7 +8311,7 @@ def gather(input, index, overwrite=True):
            output = fluid.layers.gather(x, index)
    """
    if in_dygraph_mode():
-        return core.ops.gather(input, index, None)
+        return core.ops.gather(input, index, None, 'overwrite', overwrite)
    check_variable_and_dtype(
        input, 'x',