[refactor] Re-expose important implementation classes

python/taichi/lang/__init__.py

@@ -2,9 +2,11 @@
 
 from taichi._lib import core as _ti_core
 from taichi.lang import impl
+from taichi.lang._ndarray import *
 from taichi.lang._ndrange import ndrange
 from taichi.lang.enums import Layout
 from taichi.lang.exception import *
+from taichi.lang.field import *
 from taichi.lang.impl import *
 from taichi.lang.kernel_impl import *
 from taichi.lang.matrix import *

python/taichi/lang/_ndarray.py

@@ -356,4 +356,4 @@ def setter(value):
         self.setter = setter
 
 
-__all__ = []
+__all__ = ["Ndarray", "ScalarNdarray"]

python/taichi/lang/field.py

@@ -316,4 +316,4 @@ def __init__(self, accessor, key):
         self.key = key
 
 
-__all__ = []
+__all__ = ["Field", "ScalarField"]

python/taichi/lang/matrix.py

@@ -1431,4 +1431,4 @@ def __repr__(self):
         return f'<{self.n} {self.layout} ti.Vector.ndarray>'
 
 
-__all__ = ["Matrix", "Vector"]
+__all__ = ["Matrix", "Vector", "MatrixField", "MatrixNdarray", "VectorNdarray"]

python/taichi/lang/snode.py

@@ -422,5 +422,5 @@ def get_addr(f, indices):
 
 __all__ = [
     'activate', 'append', 'deactivate', 'get_addr', 'is_active', 'length',
-    'rescale_index'
+    'rescale_index', "SNode"
 ]

python/taichi/lang/struct.py

@@ -514,4 +514,4 @@ def field(self, **kwargs):
         return Struct.field(self.members, **kwargs)
 
 
-__all__ = ["Struct"]
+__all__ = ["Struct", "StructField"]