hailin
/
sglang_v0.5.2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
sglang_v0.5.2/pytorch_2.8.0/docs/source/nn.attention.flex_attention.md

705 B
Raw Permalink Blame History 

.. 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: