[math] change the internal of surrogate function, add heaviside_p p…

…rimitive, so that all surrogate functions support JVP (forward) and VJP (backward) differentiation
brainpy · Dec 22, 2023 · 875e1bc · 875e1bc
1 parent b561f84
commit 875e1bc
Show file tree

Hide file tree

Showing 4 changed files with 1,764 additions and 6 deletions.
diff --git a/brainpy/_src/math/surrogate/__init__.py b/brainpy/_src/math/surrogate/__init__.py
@@ -2,5 +2,5 @@
 
 
 from .base import *
-from ._one_input import *
+from ._one_input_new import *
 from ._two_inputs import *