|
|
|
@ -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]:].
|
|
|
|
|