minor mod of syn

Author: ago109

ngclearn/components/synapses/hebbian/STDPSynapse.py

@@ -92,17 +92,17 @@ def __init__(self, name, shape, A_plus, A_minus, tau_plus=10., tau_minus=10.,
     def _compute_update(Aplus, Aminus, tau_plus, tau_minus, preSpike, postSpike,
                         pre_tols, post_tols, weights):
         ## calculate time deltas matrix block --> (t_post - t_pre)
-        post_m = (post_tols > 0.) * 1.
-        pre_m = (pre_tols > 0.) * 1.
+        post_m = (post_tols > 0.) #* 1.
+        pre_m = (pre_tols > 0.).T # * 1.
         t_delta = ((weights * 0 + 1.) * post_tols) - pre_tols.T ## t_delta.shape = weights.shape
-        t_delta = t_delta * post_m * pre_m.T ## mask out zero tols
+        t_delta = t_delta * post_m * pre_m ## mask out zero tols
         ## calculate post-synaptic term
-        postTerm = jnp.exp(-t_delta/tau_plus) * (t_delta > 0.)
+        postTerm = jnp.exp(-t_delta/tau_plus) * (t_delta > 0.) #* post_m * pre_m
         dWpost = postTerm * (postSpike * Aplus)
         dWpre = 0.
         if Aminus > 0.:
             ## calculate pre-synaptic term
-            preTerm = jnp.exp(-t_delta / tau_minus) * (t_delta < 0.)
+            preTerm = jnp.exp(-t_delta / tau_minus) * (t_delta < 0.) #* post_m * pre_m
             dWpre = -preTerm * (preSpike.T * Aminus)
         ## calc final weighted adjustment
         dW = (dWpost + dWpre)