设备归属

函数torch.xpu.device_of(obj)[源代码]

一个上下文管理器，用于将当前设备切换为给定对象的设备。

你可以将张量和存储都作为参数使用。如果对象没有分配在XPU上，那么这个操作不会产生任何效果。

参数

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