torch.nn.attention .. automodule:: torch.nn.attention Utils .. autosummary:: :toctree: generated :nosignatures: sdpa_kernel SDPBackend Submodules .. autosummary:: :nosignatures: bias .. toctree:: :hidden: nn.attention.bias