ott.tools.gaussian_mixture.gaussian_mixture_pair.GaussianMixturePair.get_sinkhorn#

GaussianMixturePair.get_sinkhorn(cost_matrix, **kwargs)[source]#

Get the output of Sinkhorn’s method for a given cost matrix.

Return type

SinkhornOutput

Parameters

  • cost_matrix (jax.Array) –

  • kwargs (Any) –