DŽg d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZm Z dgZ
d Z g dZ g dZ
g d Zg d
Ze egZe
egZdd Zej& j( d Z G d
de Zy) N) constraints) Distribution)
broadcast_all
lazy_propertyVonMisesc r t | }|j }|r|j | |z z }|r|S N)listpop)ycoefresults e/home/mcse/projects/flask_80/flask-venv/lib/python3.12/site-packages/torch/distributions/von_mises.py
_eval_polyr s7 :D
XXZF
a&j( M )g ?g$ @g03@g,?N?g2t?gIx?gtHZr?) e3E?g -5?gՒ+Hub?gJNYgTPÂ?g 'gZ?gUL+ߐg;^p?) ?gY ?g(z?g*O?gZ9?g.h?gӰ٩=5?) r g .kg?VmgtZOZ?g< Q g' 8`?gP⥝gqJ:N?g;PJ4qc V |dk( s|dk( sJ | dz }||z }t |t | }|dk( r| j |z }|j }d| z }| d| j z z
t |t | j z }t j | dk || }|S )zX
Returns ``log(I_order(x))`` for ``x > 0``,
where `order` is either 0 or 1.
r g @r )r _COEF_SMALLabslog _COEF_LARGEtorchwhere)xorderr smalllarger s r _log_modified_bessel_fnr C s
A:!# #
DA AA q+e, -E z% IIKE
qA
aeeg
1k%.@ A E E G GE
[[T5%
0F Mr c 6 t j |j t j | j }|j st j d|j z | j | j }|j \ }}}t j t j |z } d|| z z || z z }
|||
z
z } | d| z
z |z
dkD | |z j dz | z
dk\ z } | j r>t j | |dz
j |
j! z | }|| z }|j s|t j z | z dt j z z t j z
S )Ndtypedevice) r r r )r zerosshapeboolr$ allrandr# unbindcosmathpir anyr signacos)
loc
concentration
proposal_rr doneuu1u2u3zfcaccepts
r _rejection_sampler? Z s< ;;qwwejj DDhhj JJtagg~SYYszz JXXZ
B IIdggl #
a
JN 3 Z!^ ,A;#q(a"f\\^a-?!-Cq-HI ::< FR#XOO$5$@!DA&=D hhj
K# !dgg+ . 88r c 2 e Zd ZdZej
ej dZej
ZdZ d fd Z
d Z e d Z
e d Ze d Z ej" ej$ fd Z fd
Zed Zed Ze d
Z xZS )r aX
A circular von Mises distribution.
This implementation uses polar coordinates. The ``loc`` and ``value`` args
can be any real number (to facilitate unconstrained optimization), but are
interpreted as angles modulo 2 pi.
Example::
>>> # xdoctest: +IGNORE_WANT("non-deterministic")
>>> m = VonMises(torch.tensor([1.0]), torch.tensor([1.0]))
>>> m.sample() # von Mises distributed with loc=1 and concentration=1
tensor([1.9777])
:param torch.Tensor loc: an angle in radians.
:param torch.Tensor concentration: concentration parameter
)r3 r4 Fc t || \ | _ | _ | j j }t j
}t | ||| y r )r r3 r4 r( r Sizesuper__init__)selfr3 r4
validate_args batch_shape event_shape __class__s r rD zVonMises.__init__ sB '4S-'H$$$hhnn jjl
k=Ar c | j r| j | | j t j || j
z
z }|t
j dt j z z
t | j d z
}|S )Nr& r r )
_validate_args_validate_sampler4 r r- r3 r. r r/ r )rE valuelog_probs r rO zVonMises.log_prob sy ! !% (%% %$((2B(CC hhq477{#
$%d&8&8B
C
r c T | j j t j S r )r3 tor doublerE s r _locz
VonMises._loc s xx{{5<<((r c T | j j t j S r )r4 rQ r rR rS s r _concentrationzVonMises._concentration s !!$$U\\22r c | j }ddd|dz z z j z }|d|z j z
d|z z }d|dz z d|z z }d|z |z }t j |dk || S )Nr r& gh㈵>)rV sqrtr r )rE kappataurho _proposal_r_proposal_r_taylors r r] zVonMises._proposal_r s ##1q5!8|#))++a#g^^%%!e)436za#g. Y .{{54<);[IIr c B | j | }t j || j j | j
j }t | j | j | j | j | j
j S )a
The sampling algorithm for the von Mises distribution is based on the
following paper: D.J. Best and N.I. Fisher, "Efficient simulation of the
von Mises distribution." Applied Statistics (1979): 152-157.
Sampling is always done in double precision internally to avoid a hang
in _rejection_sample() for small values of the concentration, which
starts to happen for single precision around 1e-4 (see issue #88443).
r" ) _extended_shaper emptyrT r# r3 r$ r? rV r] rQ )rE sample_shaper( r s r samplezVonMises.sample sn $$\2 KKTYY__TXX__ M IIt**D,<,+. ." O MM - -.> ?M((//+.C . . 5 5k BM 4:c=
N N Os A.BBc | j S )z8
The provided mean is the circular one.
r3 rS s r meanz
VonMises.mean s
xxr c | j S r rk rS s r modez
VonMises.mode s xxr c dt | j d t | j d z
j z
S )z<
The provided variance is the circular one.
r rK r )r r4 exprS s r variancezVonMises.variance s>
'(:(:!D)$*<*r s + 9 B ,
~. ~. . 9 9 i
| i
r