ott.solvers.linear.solve_univariate

ott.solvers.linear.solve_univariate(geom, a=None, b=None, *, return_transport=False, return_dual_variables=False)

Solve 1D OT problems between two \(d\)-dimensional point clouds.

This function selects the underlying solver based on the following criteria:

• north_west_solver() - if return_dual_variables = True.

• uniform_solver() - if a and b are both uniform and have the same size.

• quantile_solver() - otherwise.

Parameters:

• geom (PointCloud) – Geometry containing two \(d\)-dimensional point clouds and a ground translation-invariant cost.

• a (Optional[Array]) – The first marginal. If None, it will be uniform.

• b (Optional[Array]) – The second marginal. If None, it will be uniform.

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

• return_dual_variables (bool) – Whether to also return the dual variables.

Return type:

UnivariateOutput

Returns:

The univariate output.