§
    ®˜ºh�  ã                   ó\   — d dl mZmZ d dlmZ d dlmZ d dlmZ dgZ	 G d„ de¦  «        Z
dS )é    )ÚOptionalÚUnion)ÚTensor)Úconstraints)ÚGammaÚChi2c                   óŠ   ‡ — e Zd ZdZdej        iZ	 d	deee	f         de
e         ddfˆ fd„Zd	ˆ fd„	Zedefd„¦   «         Zˆ xZS )
r   a³  
    Creates a Chi-squared distribution parameterized by shape parameter :attr:`df`.
    This is exactly equivalent to ``Gamma(alpha=0.5*df, beta=0.5)``

    Example::

        >>> # xdoctest: +IGNORE_WANT("non-deterministic")
        >>> m = Chi2(torch.tensor([1.0]))
        >>> m.sample()  # Chi2 distributed with shape df=1
        tensor([ 0.1046])

    Args:
        df (float or Tensor): shape parameter of the distribution
    ÚdfNÚvalidate_argsÚreturnc                 óV   •— t          ¦   «                              d|z  d|¬¦  «         d S )Ng      à?)r   )ÚsuperÚ__init__)Úselfr
   r   Ú	__class__s      €ú`/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/torch/distributions/chi2.pyr   zChi2.__init__   s.   ø€ õ
 	‰Œ×Ò˜˜r™ 3°mÐÑDÔDÐDÐDÐDó    c                 ó~   •— |                       t          |¦  «        }t          ¦   «                              ||¦  «        S ©N)Ú_get_checked_instancer   r   Úexpand)r   Úbatch_shapeÚ	_instanceÚnewr   s       €r   r   zChi2.expand%   s0   ø€ Ø×(Ò(­¨yÑ9Ô9ˆÝ‰wŒw�~Š~˜k¨3Ñ/Ô/Ð/r   c                 ó   — | j         dz  S )Né   )Úconcentration)r   s    r   r
   zChi2.df)   s   € àÔ! AÑ%Ð%r   r   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   ÚpositiveÚarg_constraintsr   r   Úfloatr   Úboolr   r   Úpropertyr
   Ú__classcell__)r   s   @r   r   r      sØ   ø€ € € € € ðð ð ˜[Ô1Ð2€Oð
 )-ðEð Eà�&˜%�-Ô ðEð   ”~ðEð 
ð	Eð Eð Eð Eð Eð Eð0ð 0ð 0ð 0ð 0ð 0ð ð&�Fð &ð &ð &ñ „Xð&ð &ð &ð &ð &r   N)Útypingr   r   Útorchr   Útorch.distributionsr   Útorch.distributions.gammar   Ú__all__r   © r   r   ú<module>r.      s�   ðà "Ð "Ð "Ð "Ð "Ð "Ð "Ð "à Ð Ð Ð Ð Ð Ø +Ð +Ð +Ð +Ð +Ð +Ø +Ð +Ð +Ð +Ð +Ð +ð ˆ(€ð&ð &ð &ð &ð &ˆ5ñ &ô &ð &ð &ð &r   