from .non_local import NonLocal2D

__all__ = ['NonLocal2D']