устройство

class torch.xpu.device(device) [исходный код]

Менеджер контекста, изменяющий выбранное устройство.

Параметры:

device (torch.device или int или str) – индекс устройства для выбора. Ничего не делает, если этот аргумент является отрицательным целым числом или None.