update patched synapse reset

Author: rxng8

ngclearn/components/synapses/patched/hebbianPatchedSynapse.py

@@ -276,6 +276,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 = {

ngclearn/components/synapses/patched/patchedSynapse.py

@@ -153,6 +153,15 @@ def advance_state(self):
         # Update compartment
         self.outputs.set(outputs)
 
+    @compilable
+    def reset(self, batch_size, shape):
+        preVals = jnp.zeros((batch_size, shape[0]))
+        postVals = jnp.zeros((batch_size, shape[1]))
+        inputs = preVals
+        outputs = postVals
+        self.inputs.set(inputs)
+        self.outputs.set(outputs)
+
     def save(self, directory, **kwargs):
         file_name = directory + "/" + self.name + ".npz"
         if self.bias_init != None: