设备归属

类torch.cuda.device_of(obj)[源代码]

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

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

参数

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