IsinPart

产品支持情况

产品	是否支持
Atlas A2 训练系列产品/Atlas 800I A2 推理产品	√

功能说明

• 算子功能：isin功能核心part，结合cat、sort实现isin功能

• isin算子功能：判断给定张量elements的每个元素是否在另一个张量test_elements中出现，并返回一个布尔类型张量，形状与给定张量一致。

• isin算子完整实现：

$$\begin{gathered} elementsNum=elements.length() \\ combine\_elements=cat(elements,test\_elements) \\ value,index=sort(combine\_elements) \\ z=isin\_part\_v1(value,index,elementsNum) \end{gathered}$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
value	输入	elements与test_elements拼接排序后的张量值	FLOAT、INT32	ND
index	输入	elements与test_elements拼接排序后的张量索引	INT32	ND
elementsNum	输入	elements元素数量(标量)	INT64、INT32	ND
z	输出	isin操作后的输出张量，形状由elementsNum参数决定	BOOL	ND

约束说明

无

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_isin_part_v1.cpp	通过[test_aclnn_isin_part_v1]接口方式调用IsinPartV1算子。