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.