有没有办法在 ONNX 中实现基于字符串的 swith case？

Question

希洛！

我正在尝试将分段的 logreg 模型打包到 onnx，这个想法是根据字符串值计算相应的逻辑回归，比如 if

segment = 'segm_1'

then

logreg_1.predict()

如何在 ONNX 图中创建选择器。我只找到 If 节点，但它只适用于用于分支的预制标志。如果有很多段，它看起来像不好的解决方案

我尝试使用 onnx python API 构建图形，但我发现缺少运算符