device_of

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

Менеджер контекста, который переключает текущее устройство на устройство указанного объекта.

В качестве аргументов можно использовать как тензоры, так и хранилища. Если указанный объект не размещён на XPU, ничего не происходит.

Параметры:

obj (Tensor или Storage) – объект, размещённый на выбранном устройстве.