torch.cpu

.. automodule:: torch.cpu .. currentmodule:: torch.cpu

.. autosummary:: :toctree: generated :nosignatures:

current_device
current_stream
is_available
is_initialized
synchronize
stream
set_device
device_count
StreamContext

Streams and events

.. autosummary:: :toctree: generated :nosignatures:

Stream