add tic() and toc() functions for easy coarse timing

Author: rflamary

ot/__init__.py

@@ -16,7 +16,7 @@
 from .da import sinkhorn_lpl1_mm
 
 # utils functions
-from .utils import dist, unif
+from .utils import dist, unif, tic, toc, toq
 
 __version__ = "0.1.11"
 

ot/utils.py

@@ -6,6 +6,26 @@
 from scipy.spatial.distance import cdist
 
 
+import time
+__time_tic_toc=time.time()
+
+def tic():
+    """ Python implementation of Matlab tic() function """
+    global __time_tic_toc
+    __time_tic_toc=time.time()
+
+def toc(message='Elapsed time : {} s'):
+    """ Python implementation of Matlab toc() function """
+    t=time.time()
+    print(message.format(t-__time_tic_toc))
+    return t-__time_tic_toc
+
+def toq():
+    """ Python implementation of Julia toc() function """
+    t=time.time()
+    return t-__time_tic_toc
+
+
 def kernel(x1,x2,method='gaussian',sigma=1,**kwargs):
     """Compute kernel matrix"""
     if method.lower() in ['gaussian','gauss','rbf']: