Source code for flambe.nn

# type: ignore[attr-defined]

from flambe.nn.module import Module
from flambe.nn.softmax import SoftmaxLayer
from flambe.nn.mos import MixtureOfSoftmax
from flambe.nn.embedding import Embeddings, Embedder
from flambe.nn.mlp import MLPEncoder
from flambe.nn.rnn import RNNEncoder, PooledRNNEncoder
from flambe.nn.cnn import CNNEncoder
from flambe.nn.sequential import Sequential
from flambe.nn.pooling import FirstPooling, LastPooling, SumPooling, AvgPooling, \
    StructuredSelfAttentivePooling, GeneralizedPooling
from flambe.nn.transformer import Transformer, TransformerEncoder, TransformerDecoder
from flambe.nn.transformer_sru import TransformerSRU, TransformerSRUEncoder, TransformerSRUDecoder


__all__ = ['Module', 'Embeddings', 'Embedder', 'RNNEncoder',
           'PooledRNNEncoder', 'CNNEncoder', 'MLPEncoder',
           'SoftmaxLayer', 'MixtureOfSoftmax', 'Sequential',
           'Transformer', 'TransformerEncoder', 'TransformerDecoder',
           'TransformerSRU', 'TransformerSRUEncoder', 'TransformerSRUDecoder',
           'FirstPooling', 'LastPooling', 'SumPooling', 'AvgPooling',
           'StructuredSelfAttentivePooling', 'GeneralizedPooling']