from tensor_train import *
from variables import *
from ops import *
from initializers import *
from regularizers import *
@@ -103,20 +103,41 @@
 egularizers import *
+%0Afrom shapes import *

Add shapes to init