Question

我正在尝试将一个组件的输出的最后一个元素连接到另一个组件的输入。示例如下所示：

import numpy as np
from openmdao.api import Component, Problem, Group

class C1(Component):
    def __init__(self):
        super(C1, self).__init__()
        self.add_param('fin', val=1.0)
        self.add_output('arr', val=np.zeros(5))

    def solve_nonlinear(self, params, unknowns, resids):
        fin = params['fin']
        unknowns['arr'] = np.array([2*fin])


class C2(Component):
    def __init__(self):
        super(C2, self).__init__()
        self.add_param('flt', val=0.0)
        self.add_output('fout', val=0.0)

    def solve_nonlinear(self, params, unknowns, resids):
        flt = params['flt']
        unknowns['fout'] = 2*flt

class A(Group):
    def __init__(self):
        super(A, self).__init__()

        self.add('c1', C1())
        self.add('c2', C2())

        self.connect('c1.arr[-1]', 'c2.flt')

if __name__ == '__main__':

    a = Problem()
    a.root = A()
    a.setup()

    a.run()
    print a.root.c2.unknowns['fout']

我收到错误：

openmdao.core.checks.ConnectError: Source 'c1.arr[-1]' cannot be connected to target 'c2.flt': 'c1.arr[-1]' does not exist.

有办法做到这一点吗？我知道它在旧版本的OpenMDAO中有效。

Answer 1

OpenMDAO支持使用'src_indices'arg连接到源的特定索引。例如：

self.connect('c1.arr', 'c2.flt', src_indices=[4])

目前不支持负面指数。

Answer 2

这里有一些小问题。首先，C1的solve_nonlinear方法的数组计算大小错误。它最终会解决，但你应该将数组设置为正确的大小（长度为5）。

对于数组的一部分（请参阅docs和more advanced docs），指定要连接的src_indices参数。

import numpy as np
from openmdao.api import Component, Problem, Group

class C1(Component):
    def __init__(self):
        super(C1, self).__init__()
        self.add_param('fin', val=1.0)
        self.add_output('arr', val=np.zeros(5))

    def solve_nonlinear(self, params, unknowns, resids):
        fin = params['fin']
        unknowns['arr'] = fin*np.arange(5)

class C2(Component):
    def __init__(self):
        super(C2, self).__init__()
        self.add_param('flt', val=0.0)
        self.add_output('fout', val=0.0)

    def solve_nonlinear(self, params, unknowns, resids):
        flt = params['flt']
        unknowns['fout'] = 2*flt

class A(Group):
    def __init__(self):
        super(A, self).__init__()

        self.add('c1', C1())
        self.add('c2', C2())

        self.connect('c1.arr', 'c2.flt', src_indices=[4,])

if __name__ == '__main__':

    a = Problem()
    a.root = A()
    a.setup()

    a.run()
    print a.root.c2.unknowns['fout']

将数组元素连接到另一个Component的输入

2 个答案: