def i1e(x):
x, = _promote_args_inexact("i1e", x)
return lax.bessel_i1e(x)
def i1(x):
x, = _promote_args_inexact("i1", x)
return lax.mul(lax.exp(lax.abs(x)), lax.bessel_i1e(x))
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)
