fastNLP/fastNLP/modules/aggregator/__init__.py

__all__ = ["MaxPool", "MaxPoolWithMask", "AvgPool", "MeanPoolWithMask", "KMaxPool", "Attention", "BiAttention",
           "SelfAttention"]
from .pooling import MaxPool
from .pooling import MaxPoolWithMask
from .pooling import AvgPool
from .pooling import MeanPoolWithMask
from .pooling import KMaxPool

from .attention import Attention
from .attention import BiAttention
from .attention import SelfAttention