mindspore.ops.unsorted_segment_max

mindspore.ops.unsorted_segment_max(x, segment_ids, num_segments)

分段计算输入tensor的最大值。

unsorted_segment_max的计算过程如下图所示：

$${\text{ output }}_i={\text{max}}_{j\dots }\text{ data }[j\dots ]$$

$max$ 返回元素 $j...$ 中的最大值，其中 $segment\mathrm{\_}ids[j...]==i$ 。

说明

• 如果 segment_ids 中不存在segment_id i，则输出 output[i] 将使用 x 的数据类型的最小值填充。

• segment_ids 必须是一个非负tensor。

参数：

• x (Tensor) - 输入tensor。

• segment_ids (Tensor) - 标识每个元素所属段。

• num_segments (Union[int, Tensor]) - 分段数量，可以为int或零维的tensor。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> x = mindspore.tensor([[1, 2, 3], [4, 5, 6], [4, 2, 1]])
>>> segment_ids = mindspore.tensor([0, 1, 1])
>>> num_segments = 2
>>> mindspore.ops.unsorted_segment_max(x, segment_ids, num_segments)
Tensor(shape=[2, 3], dtype=Int64, value=
[[1, 2, 3],
 [4, 5, 6]])