"""
readout layers
"""
from .linear import LinearWithIndex
from .set2set import Set2Set

__all__ = ["Set2Set", "LinearWithIndex"]

"""
readout layers
"""
from .linear import LinearWithIndex
from .set2set import Set2Set

__all__ = ["Set2Set", "LinearWithIndex"]

readout layers