SearchSorted
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | √ |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
| Atlas 200I/500 A2 推理产品 | √ |
| Atlas 推理系列产品 | √ |
| Atlas 训练系列产品 | √ |
功能说明
- 算子功能:在一个已排序的张量 sorted_sequence 中查找给定张量 values 应该插入的位置。返回与 values 相同大小的张量,其中每个元素表示给定值在原始张量中应该插入的位置。
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| sorted_sequence | 输入 | 输入 Tensor ,其最后一维的数值按升序排列。 | FLOAT16、FLOAT、INT8、INT16、INT32、INT64、UINT8 | ND |
| values | 输入 | 待插入的 Tensor ,数据类型必须与输入 sorted_sequence 相同,除了最后一维外,其余维度必须与 sorted_sequence 相同。 | FLOAT16、FLOAT、INT8、INT16、INT32、INT64、UINT8 | ND |
| sorter | 可选输入 | 可选输入 Tensor ,其shape 与未排序的 sorted_sequence 相同,包含将其在最内层维度升序排序的索引序列。 | INT64 | ND |
| dtype | 可选属性 |
• 指定输出Tensor的数据类型,仅支持 INT64 / INT32。 • 默认值为 INT64。 |
TYPE | - |
| right | 可选属性 |
• 可选布尔值,false 表示当序列中存在相同值、插入位置不唯一时,插入位置靠左对齐;true 表示在这种情况下插入位置靠右对齐。 • 默认值为 false。 |
BOOL | - |
| out | 输出 | 输出Tensor,shape与输入 values 相同。 | INT32、INT64 | ND |
约束说明
无
调用说明
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| 图模式调用 | test_geir_search_sorted | 通过算子IR构图方式调用SearchSorted算子。 |