moscot.base.problems.OTProblem.solve

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

Solve the OT problem.

Parameters:

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

• 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) – Keyword arguments for BaseSolver or its __call__ method.

Return type:

OTProblem

Returns:

: Self and updates the following fields:

• solver - the OT solver.

• solution - the OT solution.