将数组的元素连接到另一个组件的输入答案

【问题标题】：Connecting an element of array to another Component's input将数组的元素连接到另一个组件的输入
【发布时间】：2015-11-30 20:09:49
【问题描述】：

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

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。

【问题讨论】：

标签： openmdao

【解决方案1】：

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

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

目前不支持负索引。

【讨论】：

【解决方案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']

【讨论】：