устройство

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

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

Параметры:

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