device¶ class torch.mtia.device(device)[來源][來源]¶ 用於變更所選裝置的上下文管理器。 參數 device (torch.device 或 int) – 要選擇的裝置索引。如果此引數是負整數或 None，則不執行任何操作。
