Sum of Multiple Vectors

This is an example of computing the sum of a multiple vector inputs.

from csdl_om import Simulatorfrom csdl import Modelimport csdlimport numpy as np

class ExampleMultipleVector(Model):
    def define(self):        n = 3
        # Declare a vector of length 3 as input        v1 = self.declare_variable('v1', val=np.arange(n))
        # Declare another vector of length 3 as input        v2 = self.declare_variable('v2', val=np.arange(n, 2 * n))
        # Output the elementwise sum of vectors v1 and v2        self.register_output('multiple_vector_sum', csdl.sum(v1, v2))

sim = Simulator(ExampleMultipleVector())sim.run()
print('v1', sim['v1'].shape)print(sim['v1'])print('v2', sim['v2'].shape)print(sim['v2'])print('multiple_vector_sum', sim['multiple_vector_sum'].shape)print(sim['multiple_vector_sum'])

[0. 1. 2.]v2 (3,)[3. 4. 5.]multiple_vector_sum (3,)[3. 5. 7.]