From 0ed36cee828b9ee806ba3108fa82291ebd79a157 Mon Sep 17 00:00:00 2001 From: yepei6 <yepei6@huawei.com> Date: Sun, 28 Feb 2021 16:02:11 +0800 Subject: [PATCH] update the comment of TensorScatterUpdate --- mindspore/ops/operations/array_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ops/operations/array_ops.py b/mindspore/ops/operations/array_ops.py index a4af75548f..7788856f6f 100644 --- a/mindspore/ops/operations/array_ops.py +++ b/mindspore/ops/operations/array_ops.py @@ -3375,7 +3375,7 @@ class TensorScatterUpdate(PrimitiveWithInfer): Updates tensor values using given values, along with the input indices. Inputs: - - **input_x** (Tensor) - The target tensor. The dimension of input_x must be equal to indices.shape[-1]. + - **input_x** (Tensor) - The target tensor. The dimension of input_x must be no less than indices.shape[-1]. - **indices** (Tensor) - The index of input tensor whose data type is int32 or int64. - **update** (Tensor) - The tensor to update the input tensor, has the same type as input, and update.shape = indices.shape[:-1] + input_x.shape[indices.shape[-1]:].