update hebbian synapse reset bug

Author: rxng8

ngclearn/components/synapses/hebbian/hebbianSynapse.py

@@ -253,6 +253,17 @@ def evolve(self):
         self.dWeights.set(dWeights)
         self.dBiases.set(dBiases)
 
+    @compilable
+    def reset(self, batch_size, shape):
+        preVals = jnp.zeros((batch_size, shape[0]))
+        postVals = jnp.zeros((batch_size, shape[1]))
+        self.inputs.set(preVals) # inputs
+        self.outputs.set(postVals) # outputs
+        self.pre.set(preVals) # pre
+        self.post.set(postVals) # post
+        self.dWeights.set(jnp.zeros(shape)) # dW
+        self.dBiases.set(jnp.zeros(shape[1])) # db
+
     @classmethod
     def help(cls): ## component help function
         properties = {