设备 ¶

类 torch.cuda.device_of(obj)[来源][来源] ¶

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

您可以使用张量和存储作为参数。如果给定对象未分配在 GPU 上，则此操作不执行任何操作。

参数:

obj（张量或存储）- 在所选设备上分配的对象。