Module device 

Device abstraction for CPU and CUDA.

Functions

auto_device

Returns the best available device.

cpu

Returns the default CPU device.

cuda

Returns a CUDA device for the given GPU index.