20 lines (18 loc) · 341 Bytes

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

torch.nn.attention.varlen

.. automodule:: torch.nn.attention.varlen
.. currentmodule:: torch.nn.attention.varlen

.. autofunction:: varlen_attn

.. autofunction:: varlen_attn_out

.. autoclass:: AuxRequest