device_of

class torch.xpu.device_of(obj)

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

您可以將 tensors 和 storages 作為引數。如果給定的物件未分配到 XPU 上，則此操作無效。

引數

obj (Tensor 或 Storage) – 分配到選定裝置上的物件。