moscot.backends.ott.output.OTTNeuralOutput.to

OTTNeuralOutput.to(device=None)[source]

Transfer the output to another device or change its data type.

Parameters:

device (Union[Literal['cpu', 'gpu', 'tpu'], str, None]) – If not None, the output will be transferred to device.

Return type:

OTTNeuralOutput

Returns:

: The output on a saved on device.