fastNLP/fastNLP/modules/encoder/__init__.py

"""
.. todo::
    doc
"""

__all__ = [
    # "BertModel",

    "ConvolutionCharEncoder",
    "LSTMCharEncoder",

    "ConvMaxpool",

    "LSTM",

    "StarTransformer",

    "TransformerEncoder",

    "VarRNN",
    "VarLSTM",
    "VarGRU",

    "MaxPool",
    "MaxPoolWithMask",
    "AvgPool",
    "AvgPoolWithMask",

    "MultiHeadAttention",
    "BiAttention",
    "SelfAttention",
]

from .attention import MultiHeadAttention, BiAttention, SelfAttention
from .bert import BertModel
from .char_encoder import ConvolutionCharEncoder, LSTMCharEncoder
from .conv_maxpool import ConvMaxpool
from .lstm import LSTM
from .pooling import MaxPool, MaxPoolWithMask, AvgPool, AvgPoolWithMask
from .star_transformer import StarTransformer
from .transformer import TransformerEncoder
from .variational_rnn import VarRNN, VarLSTM, VarGRU