diff options
Diffstat (limited to 'ot/utils.py')
-rw-r--r-- | ot/utils.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/ot/utils.py b/ot/utils.py index c154f99..f9911a1 100644 --- a/ot/utils.py +++ b/ot/utils.py @@ -49,6 +49,12 @@ def kernel(x1, x2, method='gaussian', sigma=1, **kwargs): return K +def laplacian(x): + """Compute Laplacian matrix""" + L = np.diag(np.sum(x, axis=0)) - x + return L + + def unif(n): """ return a uniform histogram of length n (simplex) |