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 注意: