torch.cpu

创建于：2023 年 7 月 11 日 | 最后更新于：2023 年 10 月 11 日

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

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

流和事件

流

注