moscot.base.problems.OTProblem.solve

OTProblem.solve(backend='ott', solver_name=None, device=None, **kwargs)

Solve the OT problem.

Parameters:

• backend (Literal['ott']) – Which backend to use, see get_available_backends().

• solver_name (Optional[str]) – Literal defining the solver. If None, automatically infers the discrete OT solver.

• device (Union[Literal['cpu', 'gpu', 'tpu'], str, None]) – Transfer the solution to a different device, see to(). If None, keep the output on the original device.

• **kwargs (Any) – The description is missing.

Return type:

OTProblem

Returns:

: Self and updates the following fields:

• solver - the OT solver.

• solution - the OT solution.