UnsortedSegmentMin

产品支持情况

产品	是否支持
Ascend 950PR&950DT系列产品	√
Atlas A3系列产品	×
Atlas A2系列产品	×
Atlas 200I/500 A2推理产品	×
Atlas推理系列产品	×
Atlas训练系列产品	×

功能说明

• 算子功能：分段计算输入tensor的最小值。

• 计算公式：

  $$output[i]={\min }_{j...}data[j...]$$

  min 返回元素 j... 中的最小值，其中 segment_ids[j...]==i。

• 用例：

  输入tensor $data=\left[\begin{array}{ccc}[1& 2& 3]\\ [4& 5& 6]\\ [4& 2& 1]\end{array}\right]$, 分段索引tensor $segment\_ids=[0,1,1]$, $num\_segments=2$，

  输出tensor $output=\left[\begin{array}{ccc}[1& 2& 3]\\ [4& 2& 1]\end{array}\right]$

  • segment_ids不需要排序；分段索引，指示当前分段的值归属于哪个段
  • segment_ids不需要覆盖整个有效范围内所有值
  • segment_ids如果给定ID i为空，则输出data类型的最大值
  • segment_ids如果给定ID i为负值，则该值将被删除并且不会参与最小值计算
  • num_segments是scalar，代表输出的分段个数，应该大于等于实际分段的个数

参数说明

类型	名称	描述	支持的数据类型	数据格式
输入	data	输入数据	FLOAT32、INT32、INT64、BFLOAT16、FLOAT16、UINT32、UINT64	ND
输入	segment_ids	分段索引	INT32、INT64	ND
输入	num_segments	分段个数	INT32、INT64	ND
输出	value	输出值信息	FLOAT32、INT32、INT64、BFLOAT16、FLOAT16、UINT32、UINT64	ND

约束说明

data​：

• 维度至少 1（rank >=1）。

​segment_ids​：

• 必须是 INT32 或 INT64 类型。
• 形状必须是 data.shape 的前缀。
• 值必须 >= 0 且 < num_segments。

num_segments​：

• 必须是 INT32 或 INT64 类型，且 > 0。
• 应等于或大于最大 segment_id + 1。

调用说明

调用方式	样例代码	说明
图模式调用	test_geir_unsorted_segment_min	通过算子IR构图方式调用UnsortedSegmentMin算子。