Add tfp.math.owens_t for the Owen's T function.

PiperOrigin-RevId: 354462233
mountainscaler · Jan 29, 2021 · abc8624 · abc8624
1 parent 09205a1
commit abc8624
Show file tree

Hide file tree

Showing 3 changed files with 593 additions and 0 deletions.
diff --git a/tensorflow_probability/python/math/__init__.py b/tensorflow_probability/python/math/__init__.py
@@ -72,6 +72,7 @@
 from tensorflow_probability.python.math.special import lbeta
 from tensorflow_probability.python.math.special import log_gamma_correction
 from tensorflow_probability.python.math.special import log_gamma_difference
+from tensorflow_probability.python.math.special import owens_t
 from tensorflow_probability.python.math.special import round_exponential_bump_function
 from tensorflow_probability.python.random import rademacher as random_rademacher
 from tensorflow_probability.python.random import rayleigh as random_rayleigh
@@ -123,6 +124,7 @@
     'minimize',
     'MinimizeTraceableQuantities',
     'ode',
+    'owens_t',
     'pivoted_cholesky',
     'psd_kernels',
     'random_rademacher',