update documentation in bregman

rflamary · rflamary · commit 94eb4a24fe09 · 2018-05-11T16:07:03.000+02:00
diff --git a/ot/bregman.py b/ot/bregman.py
@@ -844,6 +844,8 @@ def barycenter(A, M, reg, weights=None, numItermax=1000,
         loss matrix   for OT
     reg : float
         Regularization term >0
+    weights : np.ndarray (n,)
+        Weights of each histogram i_i on the simplex        
     numItermax : int, optional
         Max number of iterations
     stopThr : float, optional
