implement sequential

Mentioned in #21
rwth-i6 · Oct 28, 2021 · 9d3b5f0 · 9d3b5f0
1 parent acf27c5
commit 9d3b5f0
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/nn/container.py b/nn/container.py
@@ -3,7 +3,7 @@
 """
 
 from __future__ import annotations
-from .base import Module, ILayerMaker, LayerRef
+from .base import Module, ILayerMaker, LayerRef, Layer
 from typing import Iterable, Iterator, Union, Dict, Callable
 
 
@@ -81,6 +81,13 @@ def forward(self, inp) -> LayerRef:
     return inp
 
 
+def sequential(source: LayerRef, *modules) -> Layer:
+  """
+  Wraps ``Sequential(*modules)(source)``
+  """
+  return Sequential(*modules)(source)
+
+
 def _convert_to_maker(obj: _ModT) -> ILayerMaker:
   if isinstance(obj, ILayerMaker):
     return obj