Merge pull request #113 from adtzlr/add-math-special-erf

Add the (Gauss) error function `math.special.erf(z)`

src/tensortrax/__about__.py

@@ -2,4 +2,4 @@
 tensorTRAX: Math on (Hyper-Dual) Tensors with Trailing Axes.
 """
 
-__version__ = "0.24.0"
+__version__ = "0.25.0"

src/tensortrax/math/special/__init__.py

@@ -5,6 +5,7 @@
 from ._special_tensor import (
     ddot,
     dev,
+    erf,
     from_triu_1d,
     from_triu_2d,
     sym,
@@ -17,6 +18,7 @@
 __all__ = [
     "ddot",
     "dev",
+    "erf",
     "from_triu_1d",
     "from_triu_2d",
     "sym",

src/tensortrax/math/special/_special_tensor.py

@@ -4,6 +4,7 @@
 
 import numpy as np
 
+from ..._helpers import Δ, Δδ, f, δ
 from ..._tensor import Tensor
 from .. import _math_array as array
 from .._math_tensor import einsum, sqrt, stack, trace, transpose
@@ -20,6 +21,23 @@ def dev(A):
     return A - trace(A) / dim * array.eye(A)
 
 
+def erf(z):
+    "The (Gauss) error function."
+    from scipy.special import erf
+
+    if isinstance(z, Tensor):
+        derf = 2 / np.sqrt(np.pi) * np.exp(-f(z) ** 2)
+        return Tensor(
+            x=erf(f(z)),
+            δx=derf * δ(z),
+            Δx=derf * Δ(z),
+            Δδx=-2 * f(z) * derf * δ(z) * Δ(z) + derf * Δδ(z),
+            ntrax=z.ntrax,
+        )
+    else:
+        return erf(z)
+
+
 def sym(A):
     "Symmetric part of a Tensor."
     return (A + transpose(A)) / 2

tests/test_math.py

@@ -69,6 +69,7 @@ def test_math():
         tm.ravel,
         tm.abs,
         tm.sign,
+        tm.special.erf,
     ]:
         assert np.allclose(fun(F), fun(T).x)