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 /docs | |
| 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 'docs')
0 files changed, 0 insertions, 0 deletions