torch.nn.attention.sdpa\_kernel =============================== .. currentmodule:: torch.nn.attention .. autofunction:: sdpa_kernel