目錄
捷徑

torch.mtia.set_device

torch.mtia.set_device(device)[來源][來源]

設定目前的裝置。

參數

device (torch.device 或 int) – 選取的裝置。如果此引數為負數，則此函數不執行任何操作。

文件

存取 PyTorch 的完整開發者文件

檢視文件

教學課程

取得適合初學者和進階開發者的深入教學課程

檢視教學課程

資源

尋找開發資源並獲得問題解答

檢視資源

為了分析流量並優化您的體驗，我們會在網站上使用 Cookie。 點擊或瀏覽即表示您同意我們使用 Cookie。 作為本網站當前的維護者，Facebook 的 Cookie 政策適用。 了解更多資訊，包括關於可用的控制項： Cookie 政策。