define functions to expose

kornia · Aug 28, 2018 · 0d84f55 · 0d84f55
1 parent a3c2494
commit 0d84f55
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/torchgeometry/__init__.py b/torchgeometry/__init__.py
@@ -1,3 +1,4 @@
+
 __version__ = '0.0.1'  # the current version of the lib
 
 from .functional import *
diff --git a/torchgeometry/functional.py b/torchgeometry/functional.py
@@ -1,5 +1,8 @@
 import torch
 
+__all__ = ["pi", "rad2deg", "deg2rad", "convert_points_from_homogeneous",
+           "convert_points_to_homogeneous", "transform_points", "inverse"]
+
 
 """Constant with number pi
 """