是否可以在 Pyomo 的目标 function 内编写非线性分段 function？

Question

我想使用 Pyomo 来解决使用 Ipopt 的非线性优化。 在这方面，我在我的代码中设计了一个 function 作为目标 function，它本身在其中使用了另一个 function。 内部 function 是非线性和分段的，定义如下。

def U3(x,omega,U_alpha3):
       if x<=omega/U_alpha3:
              return omega*x-U_alpha3/2*x**2
       else:
              return 0.5*omega**2/U_alpha3

但是这个 function 导致这个错误：

TypeError: '<=' not supported between instances of 'IndexedComponent_slice' and 'numpy.float64'

考虑到我的决策变量在 x 内部（它是“IndexedComponent_slice”类型）并且 omega 和 U_alpha3 是固定参数（它们是“numpy.float64”类型）

Answer 1

是的。 Ipopt 的 Pyomo NL 文件接口通过Expr_If表达式 class 支持这一点，考虑到关于 function 的标准警告，您定义的是两次连续可微分等（我们不会为您验证）。