torch.nn.attention

此模組包含會變更 torch.nn.functional.scaled_dot_product_attention 行為的函式和類別

公用程式

sdpa_kernel	用來選擇要用於縮放點積注意力的後端的內容管理員。
SDPBackend	一個類似枚舉的類別，包含縮放點積注意力（scaled dot product attention）的不同後端。

子模組

flex_attention	這個模組實作了 PyTorch 中 flex_attention 的使用者介面 API。
bias	定義與 scaled_dot_product_attention 一起使用的 bias 子類別
experimental