.. role:: hidden
    :class: hidden-section

torch.nn.attention
==================

.. automodule:: torch.nn.attention

Utils
-------------------
.. autosummary::
    :toctree: generated
    :nosignatures:

    sdpa_kernel
    SDPBackend

Submodules
----------
.. autosummary::
    :nosignatures:

    flex_attention
    bias

.. toctree::
    :hidden:

    nn.attention.flex_attention
    nn.attention.bias