ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.get_normalized_sinkhorn_coupling#
- GaussianMixturePair.get_normalized_sinkhorn_coupling(sinkhorn_output)[source]#
Get the normalized coupling matrix for the specified Sinkhorn output.
- Parameters:
sinkhorn_output (
SinkhornOutput
) – Sinkhorn algorithm output as returned byget_sinkhorn()
.- Return type:
- Returns:
A coupling matrix that tells how much of the mass of each component of
gmm0
is mapped to each component ofgmm1
.