コード例 #1
0
ファイル: special.py プロジェクト: GregCT/jax 
def i1e(x):
    x, = _promote_args_inexact("i1e", x)
    return lax.bessel_i1e(x)
コード例 #2
0
ファイル: special.py プロジェクト: GregCT/jax 
def i1(x):
    x, = _promote_args_inexact("i1", x)
    return lax.mul(lax.exp(lax.abs(x)), lax.bessel_i1e(x))
コード例 #3
0
ファイル: directional.py プロジェクト: tbsexton/numpyro 
 def variance(self):
     """ Computes circular variance of distribution """
     return jnp.broadcast_to(
         1. - lax.bessel_i1e(self.concentration) /
         lax.bessel_i0e(self.concentration), self.batch_shape)