device_of¶ class torch.xpu.device_of(obj)[原始碼][原始碼]¶ 將當前裝置更改為給定物件所屬裝置的上下文管理器。 您可以將張量(tensor)和儲存(storage)作為引數使用。如果給定物件未分配在 XPU 上,則此操作為空操作(no-op)。 引數 obj (Tensor 或 Storage) – 在所選裝置上分配的物件。