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	选择给定流的上下文管理器。

流和事件

流

注意。