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

流和事件¶

注意