ott.solvers.linear.solve_semidiscrete

ott.solvers.linear.solve_semidiscrete(geom, b=None, rng=None, **kwargs)

Solve a (regularized) semidiscrete OT problem.

Parameters:

• geom (SemidiscretePointCloud) – Semidiscrete geometry.

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

• rng (Optional[Array]) – Random key used for seeding.

• kwargs (Any) – Keyword arguments for SemidiscreteSolver.

Return type:

SemidiscreteOutput

Returns:

The semidiscrete output.