ott.solvers.was_solver.WassersteinSolver#

class ott.solvers.was_solver.WassersteinSolver(epsilon=None, rank=- 1, linear_ot_solver=None, min_iterations=5, max_iterations=50, threshold=0.001, store_inner_errors=False, **kwargs)[source]#

A generic solver for problems that use a linear reg-OT pb in inner loop.

Methods

Attributes

is_low_rank

Whether the solver is low-rank.

Parameters