Revision 80e01e5f6d2327976a26e467707a5ebe4018e7b1 authored by Jean Kossaifi on 07 October 2021, 16:17:20 UTC, committed by Jean Kossaifi on 07 October 2021, 21:17:07 UTC
1 parent fa27251
__init__.py
__version__ = '0.6.1'
import sys
from .base import unfold, fold
from .base import tensor_to_vec, vec_to_tensor
from .base import partial_unfold, partial_fold
from .base import partial_tensor_to_vec, partial_vec_to_tensor
from .cp_tensor import (cp_to_tensor, cp_to_unfolded,
cp_to_vec, unfolding_dot_khatri_rao,
cp_norm, cp_mode_dot, cp_normalize,
validate_cp_rank)
from .tucker_tensor import (tucker_to_tensor, tucker_to_unfolded,
tucker_to_vec, tucker_mode_dot,
validate_tucker_rank)
from .tt_tensor import tt_to_tensor, tt_to_unfolded, tt_to_vec, validate_tt_rank
from .tt_matrix import (tt_matrix_to_tensor, tt_matrix_to_tensor, validate_tt_matrix_rank,
tt_matrix_to_unfolded, tt_matrix_to_vec, tt_matrix_to_matrix)
from .tr_tensor import tr_to_tensor, tr_to_unfolded, tr_to_vec, validate_tr_rank
from .backend import (set_backend, get_backend,
#backend_context,
# backend_manager,
# _get_backend_dir, _get_backend_method,
)
# from . import backend as backend_manager
from .backend import (context, tensor, is_tensor, shape, ndim, to_numpy, copy,
concatenate, reshape, transpose, moveaxis, arange, ones,
zeros, zeros_like, eye, where, clip, max, min, argmax,
argmin, all, mean, sum, prod, sign, abs, sqrt, norm, dot,
kron, solve, lstsq, qr, kr, partial_svd, stack, eps, finfo,
matmul, index_update, check_random_state, randomized_svd,
randn, randomized_range_finder, log2, sin, cos)
# Deprecated
from .cp_tensor import kruskal_to_tensor, kruskal_to_unfolded, kruskal_to_vec
from . import backend
# Add Backend functions, dynamically dispatched
def __dir__():
"""Returns the module's __dir__, including the local variables
and augmenting it with the dynamically dispatched variables from backend.
"""
static_items = list(sys.modules[__name__].__dict__.keys())
return backend.get_backend_dir() + static_items
# return _get_backend_dir() + static_items
__getattr__ = backend.__getattribute__
# override_module_dispatch(__name__,
# backend_manager.__getattribute__,
# full_dir)
# # override_module_dispatch(__name__, _get_backend_method, full_dir)
# del override_module_dispatch, full_dir#, _get_backend_method
Computing file changes ...