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