moscot.base.problems.OTProblem.solve¶
- OTProblem.solve(backend='ott', device=None, **kwargs)[source]¶
Solve the OT problem.
- Parameters:
backend (
Literal
['ott'
]) – Which backend to use, seeget_available_backends()
.device (
Union
[Literal
['cpu'
,'gpu'
,'tpu'
],str
,None
]) – Transfer the solution to a different device, seeto()
. IfNone
, keep the output on the original device.kwargs (
Any
) – Keyword arguments forBaseSolver
or its__call__
method.
- Return type:
- Returns:
: Self and updates the following fields: