xref: /aosp_15_r20/external/pytorch/docs/source/nn.attention.rst (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1.. role:: hidden
2    :class: hidden-section
3
4torch.nn.attention
5==================
6
7.. automodule:: torch.nn.attention
8
9Utils
10-------------------
11.. autosummary::
12    :toctree: generated
13    :nosignatures:
14
15    sdpa_kernel
16    SDPBackend
17
18Submodules
19----------
20.. autosummary::
21    :nosignatures:
22
23    flex_attention
24    bias
25
26.. toctree::
27    :hidden:
28
29    nn.attention.flex_attention
30    nn.attention.bias