fix the doc of scatter

python/paddle/fluid/layers/nn.py

@@ -8558,7 +8558,7 @@ def scatter_nd_add(ref, index, updates, name=None):
 
     if in_dygraph_mode():
         op = getattr(core.ops, 'scatter_nd_add')
-        return op(ref, updates, output)
+        return op(ref, index, updates)
 
     if ref.dtype != updates.dtype:
         raise ValueError("ref and updates must have same data type.")

python/paddle/fluid/tests/unittests/test_scatter_nd_op.py

@@ -18,6 +18,7 @@ import unittest
 import numpy as np
 from op_test import OpTest
 import paddle.fluid as fluid
+import paddle
 
 
 def numpy_scatter_nd(ref, index, updates, fun):
@@ -285,22 +286,21 @@ class TestScatterNdOpRaise(unittest.TestCase):
 
 
 class TestDygraph(unittest.TestCase):
-    def test_dygraph1(self):
-        paddle.disable_static()
-        index_data = np.array([[1, 1], [0, 1], [1, 3]]).astype(np.int64)
-        index = paddle.to_tensor(index_data)
-        updates = paddle.rand(shape=[3, 9, 10], dtype='float32')
-        shape = [3, 5, 9, 10]
-
-        output = paddle.scatter_nd(index, updates, shape)
-
-    def test_dygraph2(self):
-        paddle.disable_static()
-        x = paddle.rand(shape=[3, 5, 9, 10], dtype='float32')
-        updates = paddle.rand(shape=[3, 9, 10], dtype='float32')
-        index_data = np.array([[1, 1], [0, 1], [1, 3]]).astype(np.int64)
-        index = paddle.to_tensor(index_data)
-        output = paddle.scatter_nd_add(x, index, updates)
+    def test_dygraph(self):
+        with fluid.dygraph.guard(fluid.CPUPlace()):
+            index_data = np.array([[1, 1], [0, 1], [1, 3]]).astype(np.int64)
+            index = fluid.dygraph.to_variable(index_data)
+            updates = paddle.rand(shape=[3, 9, 10], dtype='float32')
+            shape = [3, 5, 9, 10]
+            output = paddle.scatter_nd(index, updates, shape)
+
+    def test_dygraph(self):
+        with fluid.dygraph.guard(fluid.CPUPlace()):
+            x = paddle.rand(shape=[3, 5, 9, 10], dtype='float32')
+            updates = paddle.rand(shape=[3, 9, 10], dtype='float32')
+            index_data = np.array([[1, 1], [0, 1], [1, 3]]).astype(np.int64)
+            index = fluid.dygraph.to_variable(index_data)
+            output = paddle.scatter_nd_add(x, index, updates)
 
 
 if __name__ == "__main__":