where,switch算子:验证和支持

[miaobyte]

算子说明

switch(tensor<int64|int32|int8|bool> cases,listtensor Xs)->(tensor Y)
cases、 X[i]、 Y 这三者的shape必须相同

where算子，复用switch
where(tensor<int64|int32|int8|bool> cases,tensor A,tensor B)->(tensor Y)

excuter

• cuda
• ompsimd

python

[miaobyte]

#55

开发验证完成