torch.nn.attention¶

此模組包含修改 torch.nn.functional.scaled_dot_product_attention 行為的函式和類

工具函式¶

用於選擇縮放點積注意力要使用哪個後端的上下文管理器。

一個類似列舉的類，包含縮放點積注意力的不同後端。

`flex_attention`	此模組實現了 PyTorch 中 flex_attention 的面向使用者 API。
`偏置`	定義了與 scaled_dot_product_attention 配合使用的偏置子類
`實驗性`