Sum of a Single Matrix along Columns

This is an example of computing the sum of a single matrix input along the columns of the matrix.

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


class ExampleSingleMatrixAlong0(Group):
    def setup(self):
        n = 3
        m = 6

        # Declare a matrix of shape 3x6 as input
        M1 = self.declare_input('M1', val=np.arange(n * m).reshape((n, m)))

        # Output the axiswise sum of matrix M1 along the columns
        self.register_output('single_matrix_sum_along_0', ot.sum(M1,
                                                                 axes=(0, )))


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

print('M1', prob['M1'].shape)
print(prob['M1'])
print('single_matrix_sum_along_0', prob['single_matrix_sum_along_0'].shape)
print(prob['single_matrix_sum_along_0'])

M1 (3, 6)
[[ 0.  1.  2.  3.  4.  5.]
 [ 6.  7.  8.  9. 10. 11.]
 [12. 13. 14. 15. 16. 17.]]
single_matrix_sum_along_0 (6,)
[18. 21. 24. 27. 30. 33.]