ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.get_normalized_sinkhorn_coupling

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 by get_sinkhorn().

Return type:

Array

Returns:

A coupling matrix that tells how much of the mass of each component of gmm0 is mapped to each component of gmm1.