torch.cpu ¬
此包实现了在 torch.cuda
中找到的抽象,以方便编写设备无关的代码。
返回当前 cpu 设备。 |
|
返回给定设备当前选定的 |
|
返回一个布尔值,表示 CPU 当前是否可用。 |
|
等待 CPU 设备上所有流中的所有内核完成。 |
|
Context-manager StreamContext 的包装器,用于选择指定的流。 |
|
设置当前设备,在 CPU 中我们不做任何事情。 |
|
返回 CPU 设备的数量(不是核心数)。 |
|
选择指定流的上下文管理器。 |
流和事件 §
N.B. |