add snntorch neuronhook bugfix #258

Author: jasonlyik

neurobench/hooks/neuron.py

@@ -67,7 +67,15 @@ def hook_fn(self, layer, input, output):
 
         """
         if self.spiking:
-            self.activation_outputs.append(output[0])
+            if (
+                self.layer.init_hidden
+                and not self.layer.output
+                and not isinstance(output, tuple)
+            ):
+                self.activation_outputs.append(output)
+            else:
+                self.activation_outputs.append(output[0])
+
             if hasattr(layer, "mem"):
                 self.post_fire_mem_potential.append(layer.mem)