fastNLP/fastNLP/modules/encoder/__init__.py

__all__ = [
    # "BertModel",
    
    "ConvolutionCharEncoder",
    "LSTMCharEncoder",
    
    "ConvMaxpool",
    
    "Embedding",
    "StaticEmbedding",
    "ElmoEmbedding",
    "BertEmbedding",
    "StackEmbedding",
    "LSTMCharEmbedding",
    "CNNCharEmbedding",
    
    "LSTM",
    
    "StarTransformer",
    
    "TransformerEncoder",
    
    "VarRNN",
    "VarLSTM",
    "VarGRU",

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

    "MultiHeadAttention",
]
from ._bert import BertModel
from .bert import BertWordPieceEncoder
from .char_encoder import ConvolutionCharEncoder, LSTMCharEncoder
from .conv_maxpool import ConvMaxpool
from .embedding import Embedding, StaticEmbedding, ElmoEmbedding, BertEmbedding, \
    StackEmbedding, LSTMCharEmbedding, CNNCharEmbedding
from .lstm import LSTM
from .star_transformer import StarTransformer
from .transformer import TransformerEncoder
from .variational_rnn import VarRNN, VarLSTM, VarGRU

from .pooling import MaxPool, MaxPoolWithMask, AvgPool, AvgPoolWithMask
from .attention import MultiHeadAttention