mindspore.ops.unsorted_segment_max ================================== .. py:function:: mindspore.ops.unsorted_segment_max(x, segment_ids, num_segments) 沿分段计算输入Tensor的最大值。 unsorted_segment_max的计算过程如下图所示: .. image:: UnsortedSegmentMax.png .. math:: \text { output }_i=\text{max}_{j \ldots} \text { data }[j \ldots] :math:`max` 返回元素 :math:`j...` 中的最大值,其中 :math:`segment\_ids[j...] == i` 。 .. note:: - 如果 `segment_ids` 中不存在segment_id `i` ,则将使用 `x` 的数据类型的最小值填充输出 `output[i]` 。 - `segment_ids` 必须是一个非负Tensor。 **参数:** - **x** (Tensor) - shape: :math:`(x_1, x_2, ..., x_R)` 。数据类型支持float16、float32或int32。 - **segment_ids** (Tensor) - shape为 :math:`(x_1)` 的1维张量,值必须是非负数。数据类型支持int32。 - **num_segments** (int) - 分段的数量。 **返回:** Tensor,若 `num_segments` 值为 `N` ,则shape为 :math:`(N, x_2, ..., x_R)` 。 **异常:** - **TypeError** - `num_segments` 不是int类型。 - **ValueError** - `segment_ids` 的维度不等于1。