dosma.to_device

dosma.to_device(input, device=Device(type='cpu'))

Move input to device. Does not copy if same device.

Adapted from sigpy.

Parameters:

• input (array) – Input.

• device (int or Device or cupy.Device) – Output device.

Returns:

Output array placed in device.

Return type:

array