moscot.base.problems.OTProblem.solve

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

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: