diff --git a/prototorch/functions/__init__.py b/prototorch/functions/__init__.py index e69de29..d69d07c 100644 --- a/prototorch/functions/__init__.py +++ b/prototorch/functions/__init__.py @@ -0,0 +1,12 @@ +"""ProtoTorch functions.""" + +from .activations import identity, sigmoid_beta, swish_beta +from .competitions import knnc, wtac + +__all__ = [ + 'identity', + 'sigmoid_beta', + 'swish_beta', + 'knnc', + 'wtac', +] diff --git a/prototorch/modules/__init__.py b/prototorch/modules/__init__.py index e69de29..fcaffe1 100644 --- a/prototorch/modules/__init__.py +++ b/prototorch/modules/__init__.py @@ -0,0 +1,7 @@ +"""ProtoTorch modules.""" + +from .prototypes import Prototypes1D + +__all__ = [ + 'Prototypes1D', +]