Adding functions to math.py issue #6948

Author: jaharvey8

pymc/math.py

@@ -32,6 +32,7 @@
 from pytensor.tensor import (
     abs,
     and_,
+    broadcast_to,
     ceil,
     clip,
     concatenate,
@@ -45,6 +46,8 @@
     erfcinv,
     erfinv,
     exp,
+    fill,
+    fill_like,
     flatten,
     floor,
     ge,
@@ -55,9 +58,12 @@
     logaddexp,
     logsumexp,
     lt,
+    matmul,
     maximum,
+    mean,
     minimum,
     neq,
+    ones,
     ones_like,
     or_,
     prod,
@@ -73,6 +79,7 @@
     tan,
     tanh,
     where,
+    zeros,
     zeros_like,
 )
 from pytensor.tensor.linalg import solve_triangular
@@ -87,6 +94,7 @@
 __all__ = [
     "abs",
     "and_",
+    "broadcast_to",
     "ceil",
     "clip",
     "concatenate",
@@ -100,6 +108,8 @@
     "erfcinv",
     "erfinv",
     "exp",
+    "fill",
+    "fill_like",
     "flatten",
     "floor",
     "ge",
@@ -110,9 +120,12 @@
     "logaddexp",
     "logsumexp",
     "lt",
+    "matmul",
     "maximum",
+    "mean",
     "minimum",
     "neq",
+    "ones",
     "ones_like",
     "or_",
     "prod",
@@ -128,6 +141,7 @@
     "tan",
     "tanh",
     "where",
+    "zeros",
     "zeros_like",
     "kronecker",
     "cartesian",