PytorchBackend.set_device#

PytorchBackend.set_device(device_index)[source]#

Context manager that sets active compute device device for operations.

Parameters:

device_indexint: Index of the device to be set as active.