Skip to content

Commit 8d7b1c6

Browse files
committed
Add entropic_partial_gromov_wasserstein function
Introduces the entropic_partial_gromov_wasserstein function to the API and exports it. This provides access to the corresponding functionality from the underlying pot.partial module.
1 parent e285ca3 commit 8d7b1c6

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

src/PythonOT.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@ export emd,
1515
empirical_sinkhorn_divergence,
1616
mm_unbalanced,
1717
entropic_partial_wasserstein,
18+
entropic_partial_gromov_wasserstein,
1819
entropic_partial_gromov_wasserstein2,
1920
partial_wasserstein,
2021
partial_wasserstein2

src/lib.jl

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -595,6 +595,10 @@ function partial_wasserstein2(a, b, M; kwargs...)
595595
return pot.partial.partial_wasserstein2(a, b, M; kwargs...)
596596
end
597597

598+
function entropic_partial_gromov_wasserstein(C1, C2, p, q, reg; kwargs...)
599+
return pot.partial.entropic_partial_gromov_wasserstein(C1, C2, p, q, reg; kwargs...)
600+
end
601+
598602
function entropic_partial_gromov_wasserstein2(C1, C2, p, q; kwargs...)
599603
return pot.partial.entropic_partial_gromov_wasserstein2(C1, C2, p, q; kwargs...)
600604
end

0 commit comments

Comments
 (0)