device_of

class torch.cuda.device_of(obj) [source]

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

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

Параметры:

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