update

brainpy · Jan 3, 2024 · b338ca1 · b338ca1
1 parent c138e73
commit b338ca1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/brainpy/_src/math/activations.py b/brainpy/_src/math/activations.py
@@ -320,7 +320,7 @@ def softplus(x, beta: float = 1., threshold: float = 20.):
 
   """
   x = x.value if isinstance(x, Array) else x
-  return jnp.where(x * beta > threshold, x, 1 / beta * jnp.logaddexp(beta * x, 0))
+  return jnp.where(x > threshold / beta, x, 1 / beta * jnp.logaddexp(beta * x, 0))
 
 
 def log_sigmoid(x):