from .generalized_attention import GeneralizedAttention
from .non_local import NonLocal2D
__all__ = ['NonLocal2D', 'GeneralizedAttention']