Browse Source

fixed doc for ScatterMax

tags/v0.5.0-beta
jiangjinsheng 5 years ago
parent
commit
a243dc9ea3
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ops/operations/array_ops.py

+ 1
- 1
mindspore/ops/operations/array_ops.py View File

@@ -2163,7 +2163,7 @@ class ScatterMax(PrimitiveWithInfer):
Tensor, has the same shape and data type as `input_x`.

Examples:
>>> input_x = Tensor(np.array([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]]), mindspore.float32)
>>> input_x = Parameter(Tensor(np.array([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]]), mindspore.float32), name="input_x")
>>> indices = Tensor(np.array([[0, 0], [1, 1]]), mindspore.int32)
>>> update = Tensor(np.ones([2, 2, 3]) * 88, mindspore.float32)
>>> scatter_max = P.ScatterMax()


Loading…
Cancel
Save