feat: new CNN, start of creating graphs

src/models/autoencoder/autoencoder.py

@@ -50,12 +50,16 @@ class AutoEncoder(Model):
         """
         x: torch.Tensor of floats
         """
+        if len(x.shape) == 2:
+            x = x.unsqueeze(1)
         return self.encoder(x)
 
     def decode(self, x: torch.Tensor) -> torch.Tensor:
         """
         x: torch.Tensor of floats
         """
+        if len(x.shape) == 2:
+            x = x.unsqueeze(1)
         return self.decoder(x)
 
     def forward(self, x: torch.LongTensor) -> torch.Tensor: