adds object __init__.py

tfjgeorge · Apr 14, 2020 · 83fe96d · 83fe96d
1 parent 366d5f5
commit 83fe96d
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/nngeometry/object/__init__.py b/nngeometry/object/__init__.py
@@ -0,0 +1,7 @@
+from .pspace import (PSpaceDense, PSpaceBlockDiag, PSpaceDiag,
+                     PSpaceLowRank, PSpaceImplicit,
+                     PSpaceKFAC, PSpaceEKFAC)
+from .vector import (PVector, FVector)
+from .fspace import (FSpaceDense,)
+from .map import (PushForwardDense, PushForwardImplicit,
+                  PullBackDense)