TTTensor class (ivy-llc#23089)

druvdub · Oct 14, 2023 · 34dbc36 · 34dbc36
1 parent 5f32990
commit 34dbc36
Show file tree

Hide file tree

Showing 6 changed files with 654 additions and 1 deletion.
diff --git a/ivy/__init__.py b/ivy/__init__.py
@@ -83,6 +83,10 @@ class CPTensor:
     pass
 
 
+class TTTensor:
+    pass
+
+
 class Device(str):
     def __new__(cls, dev_str):
         if dev_str != "":
@@ -758,7 +762,7 @@ class Node(str):
     add_ivy_container_instance_methods,
 )
 from .data_classes.nested_array import NestedArray
-from .data_classes.factorized_tensor import TuckerTensor, CPTensor
+from .data_classes.factorized_tensor import TuckerTensor, CPTensor, TTTensor
 from ivy.utils.backend import (
     current_backend,
     compiled_backends,

diff --git a/ivy/data_classes/factorized_tensor/__init__.py b/ivy/data_classes/factorized_tensor/__init__.py
@@ -1,2 +1,3 @@
 from .tucker_tensor import TuckerTensor
 from .cp_tensor import CPTensor
+from .tt_tensor import TTTensor