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 注意: