moscot.base.output.MatrixSolverOutput.to

MatrixSolverOutput.to(device=None, dtype=None)[source]

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 (DTypeLike | None)

Return type:

BaseDiscreteSolverOutput

Returns:

: Self transferred to the device.