在 OpenMDAO 中重用具有不同参数名称的外部函数

Question

我正在尝试从外部规范生成 MDAO 问题。这需要自动创建组、学科和变量。我想重用一些分析函数，但参数不同。我必须假设这些参数的名称在重用实例之间可能不同，因此我正在寻找一种方法来制定分析函数，而无需在函数的字典样式 inputs/outputs 参数和学科输入中的键之间保持必要的一致性和输出变量。

是否可以（如果可以，如何？）在以下示例中使用以下可重用函数之一 MyReusableFunction / MyReusableFunctionAlt？

import openmdao.api as om

### External information

# I can choose the format of disciplinary functions. Some alternatives:
def MyNonReusableFunction1(inputs, outputs): # <- The way it works
    # I have to use keys 'A', 'B', 'C' here
    outputs['C'] = inputs['A']*inputs['B']

def MyNonReusableFunction2(inputs, outputs): # <- The way it works
    # I have to use keys 'D', 'E', 'F' here
    outputs['F'] = inputs['D']*inputs['E']

def MyReusableFunction(x, y): # <- The way I want it to work
    return x*y

def MyReusableFunctionAlt(inputs, outputs): # <- This would also be fine
    outputs['z'] = inputs['x']*inputs['y']

# Given structure of the problem
disciplines = {
    'D1': {
        'inputs': ['A', 'B'],
        'outputs': ['C'],
        'function': MyReusableFunction}, # <- instead of MyNonReusableFunction1
    'D2': {
        'inputs': ['D', 'E'],
        'outputs': ['F'],
        'function': MyReusableFunction}, # <- instead of MyNonReusableFunction2
}

connections = [('D2.F', 'D1.B')]

### My script starts here

problem = om.Problem()

for disc_name, disc_data in disciplines.items():
    discipine = om.ExplicitComponent()
    discipline.compute = disc_data['function']

    for param_in in disc_data['inputs']:
        discipline.add_input(param_in, 1)
        
    for param_out in disc_data['outputs']:
        discipline.add_output(param_out, 1)

    problem.add_subsystem(disc_name, discipline)

for connection in connections:
    problem.connect(connection[0], connection[1])

Answer 1

这感觉像是在 ExecComps 中注册用户定义函数的用例。这是一个全新的功能。

http://openmdao.org/twodocs/versions/latest/features/building_blocks/components/exec_comp.html#registering-user-functions

这里有一个使用示例：

http://openmdao.org/twodocs/versions/latest/features/building_blocks/components/exec_comp.html#execcomp-example-user-function-registration

这将为您处理导数，使用复数步长或有限差分，具体取决于给定函数是否复数安全。

这是您的代码示例。它不能完美地复制存储用户函数的字典，但与重新分配计算相比，通过此路径获取它们可能更容易一些。

import openmdao.api as om

def MyReusableFunction(x, y): # <- The way I want it to work
    return x*y

connections = [('D2.F', 'D1.B')]

problem = om.Problem()

om.ExecComp.register('myfunc', MyReusableFunction, complex_safe=True)

D1 = om.ExecComp('C = myfunc(A, B)')
D2 = om.ExecComp('F = myfunc(D, E)')

problem.model.add_subsystem('D1', D1)
problem.model.add_subsystem('D2', D2)

for connection in connections:
    problem.model.connect(connection[0], connection[1])

problem.setup()

problem.run_model()

如果这对您的用例不起作用，请告诉我们。

在 OpenMDAO 中重用具有不同参数名称的外部函数

Reusing external functions with different argument names in OpenMDAO

openmdao