|
Modules |
|
============= |
|
|
|
Core Modules |
|
------------ |
|
|
|
.. autoclass:: onmt.modules.Embeddings |
|
:members: |
|
|
|
|
|
Encoders |
|
--------- |
|
|
|
.. autoclass:: onmt.encoders.EncoderBase |
|
:members: |
|
|
|
.. autoclass:: onmt.encoders.MeanEncoder |
|
:members: |
|
|
|
.. autoclass:: onmt.encoders.RNNEncoder |
|
:members: |
|
|
|
|
|
Decoders |
|
--------- |
|
|
|
|
|
.. autoclass:: onmt.decoders.DecoderBase |
|
:members: |
|
|
|
.. autoclass:: onmt.decoders.decoder.RNNDecoderBase |
|
:members: |
|
|
|
.. autoclass:: onmt.decoders.StdRNNDecoder |
|
:members: |
|
|
|
.. autoclass:: onmt.decoders.InputFeedRNNDecoder |
|
:members: |
|
|
|
Attention |
|
---------- |
|
|
|
.. autoclass:: onmt.modules.AverageAttention |
|
:members: |
|
|
|
.. autoclass:: onmt.modules.GlobalAttention |
|
:members: |
|
|
|
|
|
|
|
Architecture: Transformer |
|
---------------------------- |
|
|
|
.. autoclass:: onmt.modules.PositionalEncoding |
|
:members: |
|
|
|
.. autoclass:: onmt.modules.position_ffn.PositionwiseFeedForward |
|
:members: |
|
|
|
.. autoclass:: onmt.encoders.TransformerEncoder |
|
:members: |
|
|
|
.. autoclass:: onmt.decoders.TransformerDecoder |
|
:members: |
|
|
|
.. autoclass:: onmt.modules.MultiHeadedAttention |
|
:members: |
|
:undoc-members: |
|
|
|
|
|
Architecture: Conv2Conv |
|
---------------------------- |
|
|
|
(These methods are from a user contribution |
|
and have not been thoroughly tested.) |
|
|
|
|
|
.. autoclass:: onmt.encoders.CNNEncoder |
|
:members: |
|
|
|
|
|
.. autoclass:: onmt.decoders.CNNDecoder |
|
:members: |
|
|
|
.. autoclass:: onmt.modules.ConvMultiStepAttention |
|
:members: |
|
|
|
.. autoclass:: onmt.modules.WeightNormConv2d |
|
:members: |
|
|
|
Architecture: SRU |
|
---------------------------- |
|
|
|
.. autoclass:: onmt.models.sru.SRU |
|
:members: |
|
|
|
|
|
Copy Attention |
|
-------------- |
|
|
|
.. autoclass:: onmt.modules.CopyGenerator |
|
:members: |
|
|
|
|
|
Structured Attention |
|
------------------------------------------- |
|
|
|
.. autoclass:: onmt.modules.structured_attention.MatrixTree |
|
:members: |
|
|