Average of a Single Vector

This is an example of computing the average of a single vector input.

from openmdao.api import Problem
from omtools.api import Group
import omtools.api as ot
import numpy as np


class ExampleSingleVector(Group):
    def setup(self):
        n = 3

        # Declare a vector of length 3 as input
        v1 = self.declare_input('v1', val=np.arange(n))

        # Output the average of all the elements of the vector v1
        self.register_output('single_vector_average', ot.average(v1))


prob = Problem()
prob.model = ExampleSingleVector()
prob.setup(force_alloc_complex=True)
prob.run_model()

print('v1', prob['v1'].shape)
print(prob['v1'])
print('single_vector_average', prob['single_vector_average'].shape)
print(prob['single_vector_average'])

v1 (3,)
[0. 1. 2.]
single_vector_average (1,)
[1.]