mindspore.ops.tensor_scatter_max
- mindspore.ops.tensor_scatter_max(input_x, indices, updates)[source]
Return a new tensor by performing a maximum update on input_x at the specified indices with the given update values.
Note
On GPU, if some values of the indices are out of bound, instead of raising an index error, the corresponding updates will not be updated to self tensor.
On CPU, if some values of the indices are out of bound, raising an index error.
On Ascend, out of bound checking is not supported, if some values of the indices are out of bound, unknown errors may be caused.
- Parameters
- Returns
Tensor
- Supported Platforms:
GPU
CPU
Examples
>>> import mindspore >>> input_x = mindspore.tensor([[1, 2, 3], [4, 5, 6]]) >>> indices = mindspore.tensor([[0, 0], [1, 1]]) >>> updates = mindspore.tensor([5, 5]) >>> mindspore.ops.tensor_scatter_max(input_x, indices, updates) Tensor(shape=[2, 3], dtype=Int64, value= [[5, 2, 3], [4, 5, 6]])