File: nn_modules.rst

package info (click to toggle)
pytorch-text 0.14.1-2
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 11,560 kB
  • sloc: python: 14,197; cpp: 2,404; sh: 214; makefile: 20
file content (29 lines) | stat: -rw-r--r-- 589 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
.. role:: hidden
    :class: hidden-section

torchtext.nn
=======================================

.. automodule:: torchtext.nn
.. currentmodule:: torchtext.nn

:hidden:`MultiheadAttentionContainer`
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. autoclass:: MultiheadAttentionContainer
    :members:
    :special-members: __init__

:hidden:`InProjContainer`
~~~~~~~~~~~~~~~~~~~~~~~~~

.. autoclass:: InProjContainer
    :members:
    :special-members: __init__

:hidden:`ScaledDotProduct`
~~~~~~~~~~~~~~~~~~~~~~~~~~

.. autoclass:: ScaledDotProduct
    :members:
    :special-members: __init__