device

class tensorplay.cuda.device(device: Any)[source]

Context-manager that changes the selected device.

Parameters:

device (tensorplay.Device or int) – device index to select. It’s a no-op if this argument is a negative integer or None.