torch.cpu

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

`current_stream`	Возвращает текущий выбранный `Stream` для данного устройства.
`is_available`	Возвращает значение булевого типа, указывающее, доступна ли в данный момент ЦП.
`synchronize`	Ожидает завершения всех ядер во всех потоках на устройстве ЦП.
`stream`	Обёртка над менеджером контекста StreamContext, который выбирает заданный поток.
`device_count`	Возвращает количество устройств ЦП (а не ядер).
`StreamContext`	Менеджер контекста, который выбирает заданный поток.

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

Примечание:

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