j 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 ZdS ) N) constraints) Distribution)
broadcast_all
lazy_propertyVonMisesc t | }| }|r | | |z z }| |S N)listpop)ycoefresults NC:\Projects\Historical\venv\Lib\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 s|dk sJ | 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 g @r )r _COEF_SMALLabslog _COEF_LARGEtorchwhere)xorderr smalllarger s r _log_modified_bessel_fnr C s
A::! #
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_sampler? Z 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 8 e Zd ZdZej ej dZej ZdZ d fd Z
d Z e d Z
e d Ze d Z ej ej fd
Zd 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 FNc t || \ | _ | _ | j j }t j }t
||| d S r )r r3 r4 r( r Sizesuper__init__)selfr3 r4
validate_args batch_shape event_shape __class__s r rD zVonMises.__init__ sR '4S-'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_sampler4 r r- r3 r. r r/ r )rE valuelog_probs r rO zVonMises.log_prob s ) ! !% ( ( (% %$(2B(C(CC hq47{##
$%d&8BBB
C
r c J | j t j S r )r3 tor doublerE s r _locz
VonMises._loc s x{{5<(((r c J | j t j S r )r4 rQ r rR rS 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㈵>)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<);[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 emptyrT r# r3 r$ r? rV r] rQ )rE 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.
r3 rS s r meanz
VonMises.mean s
xr c | j S r rl rS 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 rK r )r r4 exprS s r variancezVonMises.variance sF
'(:!DDD)$*r sD + + + + + + 9 9 9 9 9 9 B B B B B B B B ,
~. ~. . 9 9 9 i
i
i
i
i
| i
i
i
i
i
r