torch.cpu

此包实现了在 torch.cuda 中发现的抽象，以促进编写设备无关的代码。

当前设备	返回当前的CPU设备。
当前流	返回给定设备的当前选定的 Stream。
is_available	返回一个布尔值，指示CPU当前是否可用。
同步	等待CPU设备上所有流中的所有内核完成。
stream	围绕上下文管理器 StreamContext 的包装器，用于选择给定的流。
设置设备	设置当前设备，在CPU中我们不做任何操作。
device_count	返回CPU设备的数量（不是核心数）。
StreamContext	选择给定流的上下文管理器。

流和事件

Stream

注意