https://github.com/tensorly/tensorly
Tip revision: 59ba570beaad978f572b1b2c5587e9966a3472e9 authored by Jean Kossaifi on 18 March 2024, 18:21:24 UTC
Merge pull request #511 from OsmanMalik/master
Merge pull request #511 from OsmanMalik/master
Tip revision: 59ba570
__init__.py
"""
The :mod:`tensorly.decomposition` module includes utilities for performing
tensor decomposition such as CANDECOMP-PARAFAC and Tucker.
"""
from ._cp import parafac, CP, RandomizedCP, randomised_parafac, sample_khatri_rao
from ._nn_cp import non_negative_parafac, non_negative_parafac_hals, CP_NN_HALS, CP_NN
from ._tucker import (
tucker,
partial_tucker,
non_negative_tucker,
non_negative_tucker_hals,
Tucker,
)
from .robust_decomposition import robust_pca
from ._tt import tensor_train, tensor_train_matrix
from ._tt import TensorTrain, TensorTrainMatrix
from ._tr import tensor_ring, tensor_ring_als, TensorRing, TensorRingALS
from ._parafac2 import parafac2, Parafac2
from ._symmetric_cp import (
symmetric_parafac_power_iteration,
symmetric_power_iteration,
SymmetricCP,
)
from ._cp_power import parafac_power_iteration, power_iteration, CPPower
from ._cmtf_als import coupled_matrix_tensor_3d_factorization
from ._constrained_cp import constrained_parafac, ConstrainedCP