Add TupleSelect layer

Signed-off-by: Emanuele Ballarin <[email protected]>

ebtorch/__init__.py

@@ -92,6 +92,7 @@
 del SolvePoissonTensor
 del SwiGLU
 del TupleDecouple
+del TupleSelect
 del SilhouetteScore
 del Concatenate
 del DuplexLinearNeck

ebtorch/nn/__init__.py

@@ -40,6 +40,7 @@
 from .architectures import SirenSine
 from .architectures import SwiGLU
 from .architectures import TupleDecouple
+from .architectures import TupleSelect
 from .architectures_resnets_dm import PreActResNet
 from .architectures_resnets_dm import WideResNet
 from .convolutional_flatten import ConvolutionalFlattenLayer

ebtorch/nn/architectures.py

@@ -60,6 +60,7 @@
     "Clamp",
     "SwiGLU",
     "TupleDecouple",
+    "TupleSelect",
     "SilhouetteScore",
     "Concatenate",
     "DuplexLinearNeck",
@@ -907,6 +908,15 @@ def forward(self, xtuple: Tuple[Tensor, ...]) -> Tuple[Tensor, ...]:
         )
 
 
+class TupleSelect(nn.Module):
+    def __init__(self, idx: int = 0) -> None:
+        super().__init__()
+        self.idx: int = idx
+
+    def forward(self, xtuple: Tuple[Tensor, ...]) -> Tensor:
+        return xtuple[self.idx]
+
+
 class SilhouetteScore(nn.Module):
     """
     Layerized computation of the Silhouette Score.

setup.py

@@ -26,7 +26,7 @@ def read(fname):
 
 setup(
     name=PACKAGENAME,
-    version="0.25.11",
+    version="0.25.12",
     author="Emanuele Ballarin",
     author_email="[email protected]",
     url="https://github.com/emaballarin/ebtorch",