blob: c6546591c3b0c669d047f1780d10fc662ede6074 [file] [log] [blame]
.. 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