Merge pull request #396 from BindsNET/hananel

Fixing dimension mismatch

Author: Hananel-Hazan

bindsnet/network/topology.py

@@ -408,7 +408,7 @@ def compute(self, s: torch.Tensor) -> torch.Tensor:
             decaying spike activation).
         """
         self.firing_rates -= self.decay * self.firing_rates
-        self.firing_rates += s.float()
+        self.firing_rates += s.float().squeeze()
 
         _, indices = F.max_pool2d(
             self.firing_rates,