mod to stdp syn

NACLab · Jul 7, 2024 · a0f6b19 · a0f6b19
1 parent d973183
commit a0f6b19
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ngclearn/components/synapses/hebbian/STDPSynapse.py b/ngclearn/components/synapses/hebbian/STDPSynapse.py
@@ -95,7 +95,7 @@ def _compute_update(Aplus, Aminus, tau_plus, tau_minus, preSpike, postSpike,
         post_m = (post_tols > 0.) * 1.
         pre_m = (pre_tols > 0.) * 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_delta = t_delta * post_m * pre_m.T
         ## calculate post-synaptic term
         postTerm = jnp.exp(-t_delta/tau_plus)
         dWpost = postTerm * (postSpike * Aplus)