diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2019-12-02 11:31:32 +0100 |
|---|---|---|
| committer | Rémi Flamary <remi.flamary@gmail.com> | 2019-12-02 11:31:32 +0100 |
| commit | a6a654de5e78dd388a793fbd26f60045b05d519c (patch) | |
| tree | a8e3049507db770892d05c7747b2bf083c2d9af8 /ot/lp/EMD_wrapper.cpp | |
| parent | 57321bd0172c97b77dfc8b14972c18d063b6dda8 (diff) | |
proper documentation and parameter
Diffstat (limited to 'ot/lp/EMD_wrapper.cpp')
| -rw-r--r-- | ot/lp/EMD_wrapper.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/ot/lp/EMD_wrapper.cpp b/ot/lp/EMD_wrapper.cpp index 2aa44c1..9be2cdc 100644 --- a/ot/lp/EMD_wrapper.cpp +++ b/ot/lp/EMD_wrapper.cpp @@ -108,7 +108,7 @@ int EMD_wrap(int n1, int n2, double *X, double *Y, double *D, double *G, int EMD_wrap_return_sparse(int n1, int n2, double *X, double *Y, double *D, - long *iG, long *jG, double *G, + long *iG, long *jG, double *G, long * nG, double* alpha, double* beta, double *cost, int maxIter) { // beware M and C anre strored in row major C style!!! @@ -202,6 +202,7 @@ int EMD_wrap_return_sparse(int n1, int n2, double *X, double *Y, double *D, cur++; } } + *nG=cur; // nb of value +1 for numpy indexing } |