torch.cpu

{eval-rst}

.. automodule:: torch.cpu

{eval-rst}

.. currentmodule:: torch.cpu

{eval-rst}

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

    current_device
    current_stream
    get_capabilities
    is_available
    is_initialized
    synchronize
    stream
    set_device
    device_count
    StreamContext

Streams and events

{eval-rst}

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

    Stream