fix dygraph gather api

fix dygraph gather api

fix dygraph gather api
a98c69b6 · Zhong Hui · GitHub · e1e666a0 · a98c69b6
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/layers/nn.py python/paddle/fluid/layers/nn.py +1 -1

未找到文件。
--- 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',