device_of

class torch.xpu.device_of(obj)

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

您可以将 tensors 和 storages 作为参数。如果给定的对象未分配到 XPU 上，则此操作无效。

参数

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