device_of

class torch.xpu.device_of(obj)

将当前设备更改为给定对象所属设备的上下文管理器。

您可以将张量（tensor）和存储（storage）作为参数使用。如果给定对象未分配在 XPU 上，则此操作为空操作（no-op）。

参数

obj (Tensor 或 Storage) – 在所选设备上分配的对象。