nn.attention.rst

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