CaseCondition

产品支持情况

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

功能说明

• 算子功能：根据三个输入值与阈值的关系，输出离散case编号。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	长度为3的一维张量，依次表示i、j、k。	INT32、INT64、UINT64	ND
algorithm	属性	当前仅支持LU。	STRING	-
y	输出	输出case编号。	INT32	ND

约束说明

• x必须为shape[3]的向量。
• algorithm仅支持LU。

调用说明

调用方式	调用样例	说明
图模式调用	test_geir_case_condition	通过算子IR构图方式调用CaseCondition算子。