t3toolbox.tucker_tensor_train.t3_stack

t3toolbox.tucker_tensor_train.t3_stack(xx, use_jax: bool = False) → t3toolbox.backend.common.typ.Tuple[t3toolbox.backend.common.typ.Tuple[t3toolbox.backend.common.NDArray], t3toolbox.backend.common.typ.Tuple[t3toolbox.backend.common.NDArray]]

Stacks an array-like structure of TuckerTensorTrains into one stacked TuckerTensorTrain.

Examples

>>> import numpy as np
>>> import t3toolbox.tucker_tensor_train as t3
>>> import t3toolbox.corewise as cw
>>> x = t3.t3_corewise_randn((14,15,16), (4,5,6), (1,3,2,1), stack_shape=(3,5))
>>> xx = x.unstack()
>>> x2 = t3.t3_stack(xx)
>>> print(cw.corewise_norm(cw.corewise_sub(x.data, x2.data)))
0.0