moscot.base.output.MatrixSolverOutput.to

MatrixSolverOutput.to(device=None, dtype=None)

Transfer self to another compute device.

Parameters:

• device (Union[Literal['cpu', 'gpu', 'tpu'], str, None]) – Device where to transfer the solver output. If None, use the default device.

• dtype (dtype[Any] | None | Type[Any] | _SupportsDType[dtype[Any]] | str | Tuple[Any, int] | Tuple[Any, SupportsIndex | Sequence[SupportsIndex]] | List[Any] | _DTypeDict | Tuple[Any, Any]) –

Return type:

BaseSolverOutput

Returns:

: Self transferred to the device.