Index

A

add() (in module ttax.ops) apply_mapping() (ttax.compile.TTEinsum method) (ttax.ops.TTEinsum method) apply_single_mapping() (in module ttax.compile) are_batches_broadcastable() (in module ttax.ops) are_shapes_equal() (in module ttax.ops)

axis_dim (ttax.base_class.TTBase property) (ttax.compile.WrappedTT property) (ttax.ops.TTBase property) (ttax.ops.WrappedTT property) (ttax.utils.TTBase property) (ttax.utils.WrappedTT property)

B

batch_loc (ttax.base_class.TTBase property) (ttax.compile.WrappedTT property) (ttax.ops.TTBase property) (ttax.ops.WrappedTT property) (ttax.utils.TTBase property) (ttax.utils.WrappedTT property)

batch_shape (ttax.base_class.TTBase property) (ttax.compile.WrappedTT property) (ttax.ops.TTBase property) (ttax.ops.WrappedTT property) (ttax.utils.TTBase property) (ttax.utils.WrappedTT property) BatchIndexing (class in ttax.base_class)

C

change_input() (ttax.compile.TTEinsum method) (ttax.ops.TTEinsum method)

compile_cumulative() (in module ttax.compile) compile_independent() (in module ttax.compile)

D

deltas_to_tangent() (in module ttax.riemannian) dtype (ttax.base_class.TTBase property) (ttax.compile.WrappedTT property) (ttax.ops.TTBase property) (ttax.ops.WrappedTT property) (ttax.utils.TTBase property) (ttax.utils.WrappedTT property)

F

flat_inner() (in module ttax.ops) full() (in module ttax.ops)

full_tt_matrix() (in module ttax.ops) full_tt_tensor() (in module ttax.ops) fuse() (in module ttax.compile)

I

is_tt_matrix (ttax.base_class.TT property) (ttax.base_class.TTMatrix property) (ttax.compile.TT property) (ttax.compile.TTMatrix property) (ttax.compile.WrappedTT property) (ttax.decompositions.TT property) (ttax.decompositions.TTMatrix property) (ttax.ops.TT property) (ttax.ops.TTMatrix property) (ttax.ops.WrappedTT property) (ttax.random_.TT property) (ttax.random_.TTMatrix property) (ttax.riemannian.TT property) (ttax.riemannian.TTMatrix property) (ttax.utils.TT property) (ttax.utils.TTMatrix property) (ttax.utils.WrappedTT property)

is_tt_matrix() (in module ttax.ops) (in module ttax.utils) is_tt_object() (in module ttax.ops) (in module ttax.utils) is_tt_tensor() (in module ttax.ops) (in module ttax.utils)

M

matmul() (in module ttax.ops) matrix() (in module ttax.random_) module ttax.base_class ttax.compile ttax.decompositions ttax.ops ttax.random_ ttax.riemannian ttax.utils

multiply() (in module ttax.ops) multiply_by_scalar() (in module ttax.ops)

N

ndim (ttax.base_class.TTBase property) (ttax.compile.WrappedTT property) (ttax.ops.TTBase property) (ttax.ops.WrappedTT property) (ttax.utils.TTBase property) (ttax.utils.WrappedTT property) num_batch_dims (ttax.base_class.TT property) (ttax.base_class.TTMatrix property) (ttax.compile.TT property) (ttax.compile.TTMatrix property) (ttax.compile.WrappedTT property) (ttax.decompositions.TT property) (ttax.decompositions.TTMatrix property) (ttax.ops.TT property) (ttax.ops.TTMatrix property) (ttax.ops.WrappedTT property) (ttax.random_.TT property) (ttax.random_.TTMatrix property) (ttax.riemannian.TT property) (ttax.riemannian.TTMatrix property) (ttax.utils.TT property) (ttax.utils.TTMatrix property) (ttax.utils.WrappedTT property)

O

orthogonalize() (in module ttax.decompositions) (in module ttax.riemannian)

P

project() (in module ttax.riemannian)

R

raw_tensor_shape (ttax.base_class.TT property) (ttax.base_class.TTMatrix property) (ttax.compile.TT property) (ttax.compile.TTMatrix property) (ttax.compile.WrappedTT property) (ttax.decompositions.TT property) (ttax.decompositions.TTMatrix property) (ttax.ops.TT property) (ttax.ops.TTMatrix property) (ttax.ops.WrappedTT property) (ttax.random_.TT property) (ttax.random_.TTMatrix property) (ttax.riemannian.TT property) (ttax.riemannian.TTMatrix property) (ttax.utils.TT property) (ttax.utils.TTMatrix property) (ttax.utils.WrappedTT property)

replace() (ttax.base_class.TT method) (ttax.base_class.TTMatrix method) (ttax.compile.TT method) (ttax.compile.TTMatrix method) (ttax.decompositions.TT method) (ttax.decompositions.TTMatrix method) (ttax.ops.TT method) (ttax.ops.TTMatrix method) (ttax.random_.TT method) (ttax.random_.TTMatrix method) (ttax.riemannian.TT method) (ttax.riemannian.TTMatrix method) (ttax.utils.TT method) (ttax.utils.TTMatrix method) resolve_i_or_ij() (ttax.compile.TTEinsum method) (ttax.ops.TTEinsum method) round() (in module ttax.decompositions)

S

shape (ttax.base_class.TT property) (ttax.base_class.TTMatrix property) (ttax.compile.TT property) (ttax.compile.TTMatrix property) (ttax.compile.WrappedTT property) (ttax.decompositions.TT property) (ttax.decompositions.TTMatrix property) (ttax.ops.TT property) (ttax.ops.TTMatrix property) (ttax.ops.WrappedTT property) (ttax.random_.TT property) (ttax.random_.TTMatrix property) (ttax.riemannian.TT property) (ttax.riemannian.TTMatrix property) (ttax.utils.TT property) (ttax.utils.TTMatrix property) (ttax.utils.WrappedTT property)

T

tangent_to_deltas() (in module ttax.riemannian) tensor() (in module ttax.random_) to_distinct_letters() (ttax.compile.TTEinsum method) (ttax.ops.TTEinsum method) to_function() (in module ttax.compile) (in module ttax.ops) to_vanilla_einsum() (ttax.compile.TTEinsum method) (ttax.ops.TTEinsum method) TT (class in ttax.base_class) (class in ttax.compile) (class in ttax.decompositions) (class in ttax.ops) (class in ttax.random_) (class in ttax.riemannian) (class in ttax.utils) tt_cores (ttax.base_class.TT attribute) (ttax.base_class.TTMatrix attribute) (ttax.compile.TT attribute) (ttax.compile.TTMatrix attribute) (ttax.compile.WrappedTT property) (ttax.decompositions.TT attribute) (ttax.decompositions.TTMatrix attribute) (ttax.ops.TT attribute) (ttax.ops.TTMatrix attribute) (ttax.ops.WrappedTT property) (ttax.random_.TT attribute) (ttax.random_.TTMatrix attribute) (ttax.riemannian.TT attribute) (ttax.riemannian.TTMatrix attribute) (ttax.utils.TT attribute) (ttax.utils.TTMatrix attribute) (ttax.utils.WrappedTT property) tt_ranks (ttax.base_class.TTBase property) (ttax.compile.WrappedTT property) (ttax.ops.TTBase property) (ttax.ops.WrappedTT property) (ttax.utils.TTBase property) (ttax.utils.WrappedTT property)

tt_tt_multiply() (in module ttax.ops) tt_vmap() (in module ttax.decompositions) (in module ttax.ops) (in module ttax.riemannian) ttax.base_class module ttax.compile module ttax.decompositions module ttax.ops module ttax.random_ module ttax.riemannian module ttax.utils module TTBase (class in ttax.base_class) (class in ttax.ops) (class in ttax.utils) TTEinsum (class in ttax.compile) (class in ttax.ops) TTMatrix (class in ttax.base_class) (class in ttax.compile) (class in ttax.decompositions) (class in ttax.ops) (class in ttax.random_) (class in ttax.riemannian) (class in ttax.utils)

U

unwrap_tt() (in module ttax.compile) (in module ttax.ops)

W

WrappedTT (class in ttax.compile) (class in ttax.ops) (class in ttax.utils)