torch.cpu

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

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

流和事件

Stream

N.B.