t3toolbox.corewise.corewise_add

t3toolbox.corewise.corewise_add(X: NDArrayTree, Y: NDArrayTree) → NDArrayTree

Add nested objects, X,Y -> X+Y.

Examples

>>> import numpy as np
>>> import t3toolbox.corewise as cw
>>> X = (np.ones(3), (1, (), np.ones(2)))
>>> Y = (2*np.ones(3), (3, (), -np.ones(2)))
>>> print(cw.corewise_add(X, Y))
(array([3., 3., 3.]), (4, (), array([0., 0.])))