t3toolbox.corewise.corewise_scale

t3toolbox.corewise.corewise_scale(X: NDArrayTree, s) → NDArrayTree

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., -1., -1.]), (-2, (), array([2., 2.])))