torch.cpu

创建于: 2023年7月11日 | 最后更新于: 2025年10月13日

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

current_device	返回 CPU 的当前设备。
current_stream	返回给定设备的当前选定的 Stream。
is_available	返回一个布尔值，指示 CPU 当前是否可用。
is_initialized	如果 CPU 已初始化，则返回 True。
synchronize	等待 CPU 设备上的所有流中的所有内核完成。
stream	StreamContext 的包装器，用于选择指定的流。
set_device	设置当前设备，在 CPU 上我们不做任何操作。
device_count	返回 CPU 设备数量（不是核心数）。
StreamContext	选择给定流的上下文管理器。

流和事件

流

注意。