add scatter_nd

Author: Luka0612

tensorlayerx/backend/ops/__init__.py

@@ -104,6 +104,7 @@
 from .load_backend import cast
 from .load_backend import transpose
 from .load_backend import gather_nd
+from .load_backend import scatter_nd
 from .load_backend import clip_by_value
 from .load_backend import split
 from .load_backend import get_tensor_shape

tensorlayerx/backend/ops/mindspore_backend.py

@@ -1081,6 +1081,10 @@ def gather_nd(params, indices, batch_dims=0):
     return op(params, indices)
 
 
+def scatter_nd(indices, updates, shape):
+    raise NotImplementedError
+
+
 class ClipGradByValue(object):
     def __init__(self, clip_min=-1, clip_max=1):
         self.min = ms.Tensor(clip_min)

tensorlayerx/backend/ops/paddle_backend.py

@@ -849,6 +849,10 @@ def gather_nd(params, indices, batch_dims=0):
     return pd.gather_nd(params, indices)
 
 
+def scatter_nd(indices, updates, shape):
+    raise NotImplementedError
+
+
 class ClipGradByValue(pd.nn.ClipGradByValue):
     def __init__(self, clip_min=-1, clip_max=1):
         super().__init__(max=clip_max, min=clip_min)

tensorlayerx/backend/ops/tensorflow_backend.py

@@ -1118,6 +1118,10 @@ def gather_nd(params, indices, batch_dims=0):
     return tf.gather_nd(params, indices, batch_dims)
 
 
+def scatter_nd(indices, updates, shape):
+    return tf.scatter_nd(indices, updates, shape)
+
+
 class ClipGradByValue(object):
     def __init__(self, clip_min=-1, clip_max=1):
         self.min = clip_min

tensorlayerx/backend/ops/torch_backend.py

@@ -910,6 +910,9 @@ def gather_nd(params, indices, batch_dims=0):
     out = torch.take(params, idx)
     return out.view(out_shape)
 
+def scatter_nd(indices, updates, shape):
+    raise NotImplementedError
+
 
 class ClipGradByValue(object):
     def __init__(self, clip_min=-1, clip_max=1):