torch.cuda.device_of Class Reference

Inheritance diagram for torch.cuda.device_of:

Public Member Functions
def	__init__ (self, obj)
Public Member Functions inherited from torch.cuda.device
def	__init__ (self, device)
def	__enter__ (self)
def	__exit__ (self, args)

Additional Inherited Members
Public Attributes inherited from torch.cuda.device
	idx
	prev_idx

Detailed Description

Definition at line 238 of file __init__.py.

---

The documentation for this class was generated from the following file:

• torch/cuda/__init__.py