Fix the maxPool batch size issue

Author: hafezgh

bindsnet/network/topology.py

@@ -813,7 +813,7 @@ def reset_state_variables(self) -> None:
         """
         super().reset_state_variables()
 
-        self.firing_rates = torch.zeros(self.source.s.shape)
+        self.firing_rates = torch.zeros(self.source.batch_size,*(self.source.s.shape[1:]))
 
 
 class MaxPool2dConnection(AbstractConnection):
@@ -901,7 +901,7 @@ def reset_state_variables(self) -> None:
         """
         super().reset_state_variables()
 
-        self.firing_rates = torch.zeros(self.source.s.shape)
+        self.firing_rates = torch.zeros(self.source.batch_size,*(self.source.s.shape[1:]))
 
 
 class MaxPoo3dConnection(AbstractConnection):
@@ -989,7 +989,7 @@ def reset_state_variables(self) -> None:
         """
         super().reset_state_variables()
 
-        self.firing_rates = torch.zeros(self.source.s.shape)
+        self.firing_rates = torch.zeros(self.source.batch_size,*(self.source.s.shape[1:]))
 
 
 class LocalConnection(AbstractConnection):