捷徑

torch.cpu

這個套件實作了 torch.cuda 中的抽象概念,以方便撰寫與裝置無關的程式碼。

current_device

傳回 CPU 的目前裝置。

current_stream

傳回給定裝置目前選取的 Stream

is_available

傳回一個布林值,表示 CPU 目前是否可用。

synchronize

等待 CPU 裝置上所有串流中的所有核心完成。

stream

Context-manager StreamContext 的包裝器,用於選取給定的串流。

set_device

設定目前的裝置,在 CPU 中我們不做任何事情。

device_count

傳回 CPU 裝置(不是核心)的數量。

StreamContext

Context-manager,用於選取給定的串流。

串流和事件

Stream

注意:

文件

存取 PyTorch 的完整開發者文件

檢視文件

教學

取得針對初學者和進階開發人員的深入教學

檢視教學

資源

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

檢視資源