corewise_scale

t3toolbox.corewise.corewise_scale(X, s)

def corewise_scale(
        X:  NDArrayTree,  # any nested tree of ints/floats/arrays
        s,                # scalar multiplier
) -> NDArrayTree:         # s*X. same tree structure as X

Scale nested objects, X,s -> s*X.

Examples

>>> import numpy as np
>>> import t3toolbox.corewise as cw
>>> X = (np.ones(3), (1, (), np.ones(2)))
>>> print(cw.corewise_scale(X, 1.5))
(array([1.5, 1.5, 1.5]), (1.5, (), array([1.5, 1.5])))

Parameters:

X (NDArrayTree)

Return type:

NDArrayTree