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，用於選擇給定流。

流和事件¶

注意