diff options
| author | Alexandre Gramfort <alexandre.gramfort@m4x.org> | 2017-07-12 23:56:27 +0200 |
|---|---|---|
| committer | Alexandre Gramfort <alexandre.gramfort@m4x.org> | 2017-07-20 14:08:02 +0200 |
| commit | 95db977e8b931277af5dadbd79eccbd5fbb8bb62 (patch) | |
| tree | 4b48e558afceddfcb640e42bb0682bdcccb02b55 /ot/gpu/da.py | |
| parent | 37ca3142a4c8c808382f5eb1c23bf198c3e4610e (diff) | |
pep8
Diffstat (limited to 'ot/gpu/da.py')
| -rw-r--r-- | ot/gpu/da.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/gpu/da.py b/ot/gpu/da.py index b05ff70..c66e755 100644 --- a/ot/gpu/da.py +++ b/ot/gpu/da.py @@ -167,7 +167,7 @@ def sinkhorn_lpl1_mm(a, labels_a, b, M_GPU, reg, eta=0.1, numItermax=10, tmpC_GPU = cudamat.empty((Nfin, nbRow)).assign(0) transp_GPU.transpose().select_columns(indices_labels[i], tmpC_GPU) majs_GPU = tmpC_GPU.sum(axis=1).add(epsilon) - cudamat.pow(majs_GPU, (p-1)) + cudamat.pow(majs_GPU, (p - 1)) majs_GPU.mult(p) tmpC_GPU.assign(0) |