Vpf nddlmZddlmZddlmZddlm Z ddl m Z m Z de de de fd Z de de de fd ZdS) )laxN)_const)promote_args_inexact)Array ArrayLikexkappareturnc Ntd||\}}t|d}tjt j|||tj|dz ztjdtjzt j |zz tj S)avon Mises log probability distribution function. JAX implementation of :obj:`scipy.stats.vonmises` ``logpdf``. The von Mises probability distribution function is given by .. math:: f(x, \kappa) = \frac{1}{2\pi I_0(\kappa)}e^{\kappa\cos x} Where :math:`I_0` is the modified Bessel function :func:`~jax.scipy.special.i0` and :math:`\kappa\ge 0`, and the distribution is normalized in the interval :math:`-\pi \le x \le \pi`. Args: x: arraylike, value at which to evaluate the PDF kappa: arraylike, distribution shape parameter Returns: array of logpdf values. See Also: :func:`jax.scipy.stats.vonmises.pdf` zvonmises.logpdfr) r _lax_constjnpwherergtcoslogpi bessel_i0enan)rr zeros ]/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/jax/_src/scipy/stats/vonmises.pylogpdfrs2""3Q > >(!U E1  $ 36%&&a(@371sv:X[XfglXmXmKmCnCn(npspw x xxcFtjt||S)avon Mises probability distribution function. JAX implementation of :obj:`scipy.stats.vonmises` ``pdf``. The von Mises probability distribution function is given by .. math:: f(x, \kappa) = \frac{1}{2\pi I_0(\kappa)}e^{\kappa\cos x} Where :math:`I_0` is the modified Bessel function :func:`~jax.scipy.special.i0` and :math:`\kappa\ge 0`, and the distribution is normalized in the interval :math:`-\pi \le x \le \pi`. Args: x: arraylike, value at which to evaluate the PDF kappa: arraylike, distribution shape parameter Returns: array of pdf values. See Also: :func:`jax.scipy.stats.vonmises.logpdf` )rexpr)rr s rpdfr4s2 5!! " ""r)jaxr jax.numpynumpyrjax._src.lax.laxrrjax._src.numpy.utilrjax._src.typingrrrrrrr%s111111444444,,,,,,,,yiy yeyyyy<#9#Y#5######r