device_of

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

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

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

Параметры:

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