device_of¶ class torch.xpu.device_of(obj)[source]¶ 上下文管理器,将当前设备更改为给定对象的设备。 您可以使用张量和存储作为参数。如果给定对象未分配在 XPU 上,则此操作是无操作。 参数 obj (Tensor 或 Storage) – 分配在所选设备上的对象。