1.. role:: hidden 2 :class: hidden-section 3 4====================================== 5torch.nn.attention.flex_attention 6====================================== 7 8.. currentmodule:: torch.nn.attention.flex_attention 9.. py:module:: torch.nn.attention.flex_attention 10.. autofunction:: flex_attention 11 12BlockMask Utilities 13------------------- 14 15.. autofunction:: create_block_mask 16.. autofunction:: create_mask 17.. autofunction:: and_masks 18.. autofunction:: or_masks 19.. autofunction:: noop_mask 20 21BlockMask 22--------- 23 24.. autoclass:: BlockMask 25 :members: 26 :undoc-members: 27