Update _event_matvec.py

brainpy · Feb 29, 2024 · 32fa658 · 32fa658
1 parent 3e85428
commit 32fa658
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/brainpy/_src/math/jitconn/_event_matvec.py b/brainpy/_src/math/jitconn/_event_matvec.py
@@ -792,6 +792,8 @@ def event_mv_prob_homo_taichi(
   """
   events = as_jax(events)
   if isinstance(weight, float): weight = as_jax(weight)
+  print(weight.shape)
+  print(weight)
   weight = jnp.atleast_1d(as_jax(weight))
   conn_len = jnp.ceil(1 / conn_prob) * 2 - 1
   conn_len = jnp.asarray(jnp.atleast_1d(conn_len), dtype=jnp.int32)