slight extension to rate-cell for tensor-shaping

NACLab · Jul 14, 2024 · 52f36ae · 52f36ae
1 parent f4b81b2
commit 52f36ae
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ngclearn/components/neurons/graded/rateCell.py b/ngclearn/components/neurons/graded/rateCell.py
@@ -205,7 +205,8 @@ def _advance_state(dt, fx, dfx, tau_m, priorLeakRate, intgFlag, priorType,
             zF = fx(z) ## post-activation function value(s)
         else:
             ## run in "stateless" mode (when no membrane time constant provided)
-            z = _run_cell_stateless(j)
+            j_total = j + j_td
+            z = _run_cell_stateless(j_total)
             zF = fx(z)
         return j, j_td, z, zF