Module device 

Device abstraction for CPU and CUDA.

Functions

auto_device

Returns the default device for the active backend.

cpu

Returns the default CPU device.

cuda

Returns an error — CUDA is not compiled in.