diff options
author | Antoine Collas <contact@antoinecollas.fr> | 2023-03-21 15:18:09 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-03-21 15:18:09 +0100 |
commit | b9ed7b1650475420cc5bbec6c31476cc098790d5 (patch) | |
tree | f6624d9509288466ac3e9c0cd475b4a984d72ce4 /RELEASES.md | |
parent | c48cd76235569ada98af6b1bba589510a2818906 (diff) |
[MRG] Make partial_wasserstein, partial_wasserstein2 and entropic_partial_wasserstein work with backend (#449)
* add test of partial_wasserstein with torch tensors
* WIP: differentiable ot.partial.partial_wasserstein
* change test of torch partial
* make partial_wasserstein2 work with torch
* test backward through ot.partial.partial_wasserstein2
* add test of entropic_partial_wasserstein with torch tensors
* make entropic_partial_wasserstein work with torch tensors
* add test of backward through entropic_partial_wasserstein
* rm unused import
* test partial_wasserstein with all backends
* tests of partial fcts: check if torch is available
* partial: check if marginals are empty arrays
* add tests when marginals are empty arrays and/or m=None
* add PR to RELEASES.md
---------
Co-authored-by: Antoine Collas <22830806+antoinecollas@users.noreply.github.com>
Co-authored-by: RĂ©mi Flamary <remi.flamary@gmail.com>
Diffstat (limited to 'RELEASES.md')
-rw-r--r-- | RELEASES.md | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/RELEASES.md b/RELEASES.md index 5d966c2..e4c6e15 100644 --- a/RELEASES.md +++ b/RELEASES.md @@ -14,7 +14,7 @@ - Added Generalized Wasserstein Barycenter solver + example (PR #372), fixed graphical details on the example (PR #376) - Added Free Support Sinkhorn Barycenter + example (PR #387) - New API for OT solver using function `ot.solve` (PR #388) -- Backend version of `ot.partial` and `ot.smooth` (PR #388) +- Backend version of `ot.partial` and `ot.smooth` (PR #388 and #449) - Added argument for warmstart of dual potentials in Sinkhorn-based methods in `ot.bregman` (PR #437) - Add parameters method in `ot.da.SinkhornTransport` (PR #440) - `ot.dr` now uses the new Pymanopt API and POT is compatible with current Pymanopt (PR #443) |