""" Modules implementing layers in pytorch by inheriting from torch.nn.Module """ from . import attention, dim_reduction, pooling