zhangz0911gm
5 years ago
1 changed files with 2 additions and 2 deletions
Unified View
Diff Options
+ 2
- 2
mindspore/ops/operations/array_ops.py
View File
| @@ -1956,7 +1956,7 @@ class ScatterNdUpdate(PrimitiveWithInfer): | |||||
| use_locking (bool): Whether protect the assignment by a lock. Default: True. | use_locking (bool): Whether protect the assignment by a lock. Default: True. | ||||
| Inputs: | Inputs: | ||||
| - **input_x** (Tensor) - The target tensor. | |||||
| - **input_x** (Parameter) - The target tensor, with data type of Parameter. | |||||
| - **indices** (Tensor) - The index of input tensor. | - **indices** (Tensor) - The index of input tensor. | ||||
| - **update** (Tensor) - The tensor to add to the input tensor, has the same type as input. | - **update** (Tensor) - The tensor to add to the input tensor, has the same type as input. | ||||
| @@ -1964,7 +1964,7 @@ class ScatterNdUpdate(PrimitiveWithInfer): | |||||
| Tensor, has the same shape and type as `input_x`. | Tensor, has the same shape and type as `input_x`. | ||||
| Examples: | Examples: | ||||
| >>> input_x = Tensor(np.array([[-0.1, 0.3, 3.6], [0.4, 0.5, -3.2]]), mindspore.float32) | |||||
| >>> input_x = mindspore.Parameter(Tensor(np.array([[-0.1, 0.3, 3.6], [0.4, 0.5, -3.2]]), mindspore.float32)) | |||||
| >>> indices = Tensor(np.array([[0, 0], [1, 1]]), mindspore.int32) | >>> indices = Tensor(np.array([[0, 0], [1, 1]]), mindspore.int32) | ||||
| >>> update = Tensor(np.array([1.0, 2.2]), mindspore.float32) | >>> update = Tensor(np.array([1.0, 2.2]), mindspore.float32) | ||||
| >>> op = P.ScatterNdUpdate() | >>> op = P.ScatterNdUpdate() | ||||