tensorly.decomposition.tensor_train

tensor_train(input_tensor, rank, svd='truncated_svd', 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

svdstr, default is ‘truncated_svd’

function to use to compute the SVD, acceptable values in tensorly.SVD_FUNS

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.