mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
2b43d339fe
# Summary Makes the prototype API flex_attention public Pull Request resolved: https://github.com/pytorch/pytorch/pull/130755 Approved by: https://github.com/Chillee
30 lines
No EOL
407 B
ReStructuredText
30 lines
No EOL
407 B
ReStructuredText
.. 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 |