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

流和事件¶

注意