cpu.rst

torch.cpu

.. automodule:: torch.cpu

.. currentmodule:: torch.cpu

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

    current_device
    current_stream
    is_available
    synchronize
    stream
    set_device
    device_count
    StreamContext

Streams and events

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

    Stream