corewise_add#

t3toolbox.corewise.corewise_add(X, Y)#
def corewise_add(
        X:  NDArrayTree,  # any nested tree of ints/floats/arrays
        Y:  NDArrayTree,  # same tree structure as X; leaves broadcast-compatible
) -> NDArrayTree:         # X+Y. same tree structure as X

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.])))
Parameters:
Return type:

NDArrayTree