ott.solvers.linear.univariate.uniform_solver

ott.solvers.linear.univariate.uniform_solver(prob, return_transport=False)

Univariate solver between two equally sized and uniformly weighted distributions.

Parameters:

• prob (LinearProblem) – Problem with two point clouds of shapes [n, d] and [n, d] and a ground cost. The [n,] sized probability weights are stored in attributes a and b.

• return_transport (bool) – Whether to also return the mapped pairs used to compute the transport_matrices.

Return type:

UnivariateOutput

Returns:

The univariate output. Note that the mass_paired_indices can be \(0\) in some entries, but always sums to \(1\) for each of the \(d\) slices.