metastable-projections/projections/__init__.py
2022-09-03 11:22:29 +02:00

6 lines
249 B
Python

#TODO: License or such
from .base_projection_layer import BaseProjectionLayer
from .frob_projection_layer import FrobeniusProjectionLayer
from .kl_projection_layer import KLProjectionLayer
from .w2_projection_layer import WassersteinProjectionLayer