torch.cpu

此包实现 中的抽象，以便于编写与设备无关的代码。torch.cuda

current_device	返回 cpu 的当前设备。
current_stream	返回当前为给定设备选择的 ID。
is_available	返回一个布尔值，指示 CPU 当前是否可用。
同步	等待 CPU 设备上所有流中的所有内核完成。
流	围绕选择给定流的 Context-manager StreamContext 的包装器。
set_device	设置当前设备，在 CPU 中我们什么都不做。
device_count	返回 CPU 设备数（不是内核数）。
StreamContext （流上下文）	选择给定流的 context-manager。

流和事件

流

注