tjk d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dlm Z m
Z
dgZ d Z g dZ
g d Zg d
Zg d Ze
egZeegZdd Zej j d
Z G d de ZdS ) N)Tensor) constraints) Distribution)
broadcast_all
lazy_propertyVonMisesc t | }| }|r | | |z z }| |S N)listpop)ycoefresults [/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/torch/distributions/von_mises.py
_eval_polyr sG ::D
XXZZF
)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 |dk r|dk rt d| | dz }||z }t |t | }|dk r| |z }| }d| z }| d| z z
t |t
| z }t
j | dk || }|S )zX
Returns ``log(I_order(x))`` for ``x > 0``,
where `order` is either 0 or 1.
r zorder must be 0 or 1, got g @r )AssertionErrorr _COEF_SMALLabslog _COEF_LARGEtorchwhere)xorderr
smalllarger s r _log_modified_bessel_fnr" D s
zzeqjj A% A ABBB
DA AA q+e, - -E zz% IIKKE
qA
aeegg
1k%.@ A A E E G G GE
[T5%
0
0F Mr c t j |j t j | j }| st j d|j z | j | j }| \ }}}t j t j |z } d|| z z || z z }
|||
z
z } | d| z
z |z
dk | |z dz | z
dk z } |
rEt j | |dz
|
z | }|| z }| |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_samplerA \ sP ;qwej D D DDhhjj
! Jtag~SYsz J J JXXZZ
B Idgl # #
a
JN 3 Z!^ ,A;#q(a"f\\^^a-?!-Cq-HI ::<< ! FR#XOO$5$5$@!DDA&=D hhjj
!
K# !dg+ . 88r c v e Zd ZdZej ej dZej ZdZ dde
de
de dz ddf fd
Z d
Z
ede
fd Zede
fd Zede
fd
Z ej ej fd Zd fd Zede
fd Zede
fd Zede
fd 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
)r5 r6 FNr5 r6
validate_argsreturnc t || \ | _ | _ | j j }t j }t
||| d S r
)r r5 r6 r* r Sizesuper__init__)selfr5 r6 rC batch_shape event_shape __class__s r rH zVonMises.__init__ sT (5S-'H'H$$$hn jll
k=AAAAAr c | j r| | | j t j || j z
z }|t
j dt j z z
t | j d z
}|S )Nr( r r )
_validate_args_validate_sampler6 r r/ r5 r0 r r1 r" )rI valuelog_probs r rR zVonMises.log_prob s ) ! !% ( ( (% %$(2B(C(CC hq47{##
$%d&8BBB
C
r c J | j t j S r
)r5 tor doublerI s r _locz
VonMises._loc s x{{5<(((r c J | j t j S r
)r6 rT r rU rV s r _concentrationzVonMises._concentration s !$$U\222r c | j }ddd|dz z z z }|d|z z
d|z z }d|dz z d|z z }d|z |z }t j |dk || S )Nr r( gh㈵>)rY sqrtr r )rI kappataurho _proposal_r_proposal_r_taylors r r` zVonMises._proposal_r s #1q5!8|#))+++a#g^^%%%!e)436za#g. Y .{54<);[IIIr c | | }t j || j j | j j }t | 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 emptyrW r% r5 r& rA rY r` rT )rI sample_shaper* r s r samplezVonMises.sample si $$\22 KTY_TX_ M M M It*D,>+.. ." O O O M - -.> ? ?M(//+..C . 5 5k B BM 4::c=
NNN N N N Os $ A3BBc | j S )z8
The provided mean is the circular one.
r5 rV s r meanz
VonMises.mean s
xr c | j S r
ro rV s r modez
VonMises.mode s xr c dt | j d t | j d z
z
S )z<
The provided variance is the circular one.
r rN r )r" r6 exprV s r variancezVonMises.variance sF
'(:!DDD)$*r sV + + + + + + 9 9 9 9 9 9 B B B B B B B B ,
~. ~. 0 9 9 9"o
o
o
o
o
| o
o
o
o
o
r