SearchSorted

产品支持情况

产品	是否支持
Ascend 950PR&950DT系列产品	√
Atlas A3系列产品	√
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

约束说明

无

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_search_sorted	通过aclnnSearchSorted接口方式调用SearchSorted算子。
aclnn调用	test_aclnn_search_sorteds	通过aclnnSearchSorteds接口方式调用SearchSorted算子。
图模式调用	test_geir_search_sorted	通过算子IR构图方式调用SearchSorted算子。