Update _event_matvec.py

brainpy · Feb 29, 2024 · 7c950bd · 7c950bd
1 parent 32fa658
commit 7c950bd
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/brainpy/_src/math/jitconn/_event_matvec.py b/brainpy/_src/math/jitconn/_event_matvec.py
@@ -794,7 +794,11 @@ def event_mv_prob_homo_taichi(
   if isinstance(weight, float): weight = as_jax(weight)
   print(weight.shape)
   print(weight)
-  weight = jnp.atleast_1d(as_jax(weight))
+  try:
+    weight = jnp.atleast_1d(as_jax(weight))
+  except:
+    print(f'weight_value:{weight}')
+    print(f'weight_shape:{weight.shape}')
   conn_len = jnp.ceil(1 / conn_prob) * 2 - 1
   conn_len = jnp.asarray(jnp.atleast_1d(conn_len), dtype=jnp.int32)
   if seed is None: