快捷方式

device_of

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

將當前裝置更改為給定物件所屬裝置的上下文管理器。

您可以將張量(tensor)和儲存(storage)作為引數使用。如果給定物件未分配在 XPU 上,則此操作為空操作(no-op)。

引數

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

文件

訪問 PyTorch 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源