mindspore.Tensor.unsorted_segment_max

Tensor.unsorted_segment_max(segment_ids, num_segments)

For details, please refer to mindspore.ops.unsorted_segment_max().