device¶ class torch.cuda.device(device)[source][source]¶ 改變所選裝置的上下文管理器。 引數 device (torch.device 或 int) – 要選擇的裝置索引。如果此引數為負整數或 None，則此操作無效。
