捷徑

torch.cpu

這個套件實現了在 torch.cuda 中找到的抽象概念,以方便編寫與設備無關的程式碼。

current_device

返回 cpu 的目前設備。

current_stream

返回給定設備的當前選定的 Stream

is_available

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

synchronize

等待 CPU 設備上所有串流中的所有核心完成。

stream

包裝選定特定串流的上下文管理器 StreamContext。

set_device

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

device_count

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

StreamContext

選擇指定 stream 的情境管理器。

Streams and events

Stream

注意:

文件

取得 PyTorch 完整的開發者文件

檢視文件

教學

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

檢視教學

資源

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

檢視資源