| .. role:: hidden | |
| :class: hidden-section | |
| torch.nn.attention | |
| ================== | |
| .. automodule:: torch.nn.attention | |
| Utils | |
| ------------------- | |
| .. autosummary:: | |
| :toctree: generated | |
| :nosignatures: | |
| sdpa_kernel | |
| SDPBackend | |
| Submodules | |
| ---------- | |
| .. autosummary:: | |
| :nosignatures: | |
| bias | |
| .. toctree:: | |
| :hidden: | |
| nn.attention.bias |