torch.cpu

創建於：2023 年 7 月 11 日 | 最後更新於：2023 年 10 月 11 日

此包實現了 torch.cuda 中提供的抽象，以方便編寫裝置無關的程式碼。

current_device	返回 CPU 的當前裝置。
current_stream	返回給定裝置當前選擇的 Stream。
is_available	返回一個布林值，指示 CPU 當前是否可用。
synchronize	等待 CPU 裝置上所有流中的所有核心完成。
stream	StreamContext 的包裝器，用於選擇給定的流。
set_device	設定當前裝置，在 CPU 上不做任何操作。
device_count	返回 CPU 裝置的數量（不是核心數）。
StreamContext	選擇給定流的上下文管理器。

流和事件

流

注意。