mindspore.ops.tensor_scatter_max

View Source On Gitee
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.

output[indices]=max(input_x,update)

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
  • input_x (Tensor) – The input tensor.

  • indices (Tensor) – The specified indices. The rank must be at least 2.

  • updates (Tensor) – The update values.

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]])