add the SimpleBaseNode

fangwei123456 · Dec 4, 2023 · ec09d18 · ec09d18
1 parent 73e67fb
commit ec09d18
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/spikingjelly/activation_based/neuron.py b/spikingjelly/activation_based/neuron.py
@@ -36,6 +36,13 @@ def __init__(self, v_threshold: float = 1., v_reset: float = 0.,
         self.step_mode = step_mode
         self.register_memory(name='v', value=0.)
 
+    def single_step_forward(self, x: torch.Tensor):
+
+        self.neuronal_charge(x)
+        spike = self.neuronal_fire()
+        self.neuronal_reset(spike)
+        return spike
+
     def neuronal_charge(self, x: torch.Tensor):
         raise NotImplementedError