tensorly.decomposition.tensor_train_matrix

tensor_train_matrix(tensor, rank)[source]

Decompose a tensor into a matrix in tt-format

Parameters:
tensortensorized matrix

if your input matrix is of size (4, 9) and your tensorized_shape (2, 2, 3, 3) then tensor should be tl.reshape(matrix, (2, 2, 3, 3))

rank‘same’, float or int tuple
  • if ‘same’ creates a decomposition with the same number of parameters as tensor
  • if float, creates a decomposition with rank x the number of parameters of tensor
  • otherwise, the actual rank to be used, e.g. (1, rank_2, …, 1) of size tensor.ndim//2. Note that boundary conditions dictate that the first rank = last rank = 1.
Returns:
tt_matrix