捷徑

device_of

class torch.xpu.device_of(obj)[原始碼][原始碼]

Context-manager,用於將目前的裝置變更為給定物件的裝置。

您可以同時使用 tensors 和 storages 作為參數。如果給定的物件未分配在 XPU 上,則此操作不起作用。

參數

obj (TensorStorage) – 分配在選定裝置上的物件。

文件

存取 PyTorch 的完整開發者文件

檢視文件

教學

取得初學者和高級開發人員的深入教學

檢視教學

資源

尋找開發資源並取得您問題的解答

檢視資源