mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
26383a6cc0
This reverts commit 92bb323d36.
Reverted https://github.com/pytorch/pytorch/pull/131073 on behalf of https://github.com/albanD due to The docs build fails here and in trunk ([comment](https://github.com/pytorch/pytorch/pull/131073#issuecomment-2242997958))
23 lines
482 B
ReStructuredText
23 lines
482 B
ReStructuredText
.. 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
|
|
|
|
BlockMask
|
|
---------
|
|
|
|
.. autoclass:: BlockMask
|
|
:members:
|
|
:undoc-members:
|