torch.cpu

Создано: 11 июля 2023 г. | Последнее обновление: 8 мая 2026 г.

Этот пакет реализует абстракции, представленные в torch.cuda, чтобы упростить написание кода, не зависящего от устройства.

`current_device`	Возвращает текущее устройство CPU.
`current_stream`	Возвращает выбранный в данный момент `Stream` для указанного устройства.
`get_capabilities`	Возвращает неизменяемое отображение возможностей CPU, обнаруженных во время выполнения.
`is_available`	Возвращает логическое значение, указывающее, доступен ли в данный момент CPU.
`is_initialized`	Возвращает True, если CPU инициализирован.
`synchronize`	Ожидает завершения всех ядер во всех потоках на устройстве CPU.
`stream`	Обёртка над контекстным менеджером StreamContext, который выбирает указанный поток.
`set_device`	Устанавливает текущее устройство; для CPU никаких действий не выполняется.
`device_count`	Возвращает количество устройств CPU (не ядер).
`StreamContext`	Контекстный менеджер, который выбирает указанный поток.

Потоки и события

Примечание.

© 2026, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://docs.pytorch.org/docs/2.14/cpu.html