From a98c69b6c65a9ed926cdb3ba2cf38dc305fa72dc Mon Sep 17 00:00:00 2001 From: Zhong Hui Date: Wed, 28 Oct 2020 18:34:32 +0800 Subject: [PATCH] fix dygraph gather api fix dygraph gather api --- python/paddle/fluid/layers/nn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py index adde9cbd19..760f5ce58b 100755 --- 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', -- GitLab