sglang_v0.5.2/pytorch_2.8.0/docs/source/nn.attention.flex_attention.md

705 B

.. role:: hidden
    :class: hidden-section

torch.nn.attention.flex_attention

.. currentmodule:: torch.nn.attention.flex_attention
.. py:module:: torch.nn.attention.flex_attention
.. autofunction:: flex_attention

BlockMask Utilities

.. autofunction:: create_block_mask
.. autofunction:: create_mask
.. autofunction:: create_nested_block_mask
.. autofunction:: and_masks
.. autofunction:: or_masks
.. autofunction:: noop_mask

BlockMask

.. autoclass:: BlockMask
    :members:
    :undoc-members: