tensorly.decomposition.tensor_train

tensor_train(input_tensor, rank, verbose=False)[source]

TT decomposition via recursive SVD

Decomposes input_tensor into a sequence of order-3 tensors (factors) – also known as Tensor-Train decomposition [1].
Parameters:
input_tensortensorly.tensor
rank{int, int list}

maximum allowable TT rank of the factors if int, then this is the same for all the factors if int list, then rank[k] is the rank of the kth factor

verboseboolean, optional

level of verbosity

Returns:
factorsTT factors

order-3 tensors of the TT decomposition

References

[1]Ivan V. Oseledets. “Tensor-train decomposition”, SIAM J. Scientific Computing, 33(5):2295–2317, 2011.