From b9ed7b1650475420cc5bbec6c31476cc098790d5 Mon Sep 17 00:00:00 2001 From: Antoine Collas Date: Tue, 21 Mar 2023 15:18:09 +0100 Subject: [MRG] Make partial_wasserstein, partial_wasserstein2 and entropic_partial_wasserstein work with backend (#449) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * 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 --- RELEASES.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'RELEASES.md') 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) -- cgit v1.2.3