mindspore.ops.tensor_scatter_min
- mindspore.ops.tensor_scatter_min(input_x, indices, updates)[源代码]
返回一个新tensor,根据指定索引和更新值对 input_x 进行最小值更新。
说明
如果 indices 中的值超出输入 input_x 索引范围:
GPU平台上相应的 updates 不会更新到 input_x 且不会抛出索引错误。
CPU平台上直接抛出索引错误。
Ascend平台不支持越界检查,若越界可能会造成未知错误。
- 参数:
input_x (Tensor) - 输入tensor。
indices (Tensor) - 指定索引,其秩至少为2。
updates (Tensor) - 更新值。
- 返回:
Tensor
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> input_x = mindspore.tensor([[1, 2, 3], [4, 5, 6]]) >>> indices = mindspore.tensor([[0, 0], [1, 1]]) >>> updates = mindspore.tensor([5, 1]) >>> mindspore.ops.tensor_scatter_min(input_x, indices, updates) Tensor(shape=[2, 3], dtype=Int64, value= [[1, 2, 3], [4, 1, 6]])