
    Ǆg                     8    d dl mZ d dlmZ dgZ G d de      Zy)    )constraints)GammaChi2c                   \     e Zd ZdZdej
                  iZd fd	Zd fd	Ze	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
    dfc                 0    t         |   d|z  d|       y )Ng      ?)validate_args)super__init__)selfr   r	   	__class__s      `/home/mcse/projects/flask_80/flask-venv/lib/python3.12/site-packages/torch/distributions/chi2.pyr   zChi2.__init__   s    r3mD    c                 P    | j                  t        |      }t        |   ||      S N)_get_checked_instancer   r
   expand)r   batch_shape	_instancenewr   s       r   r   zChi2.expand   s&    ((y9w~k3//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   propertyr   __classcell__)r   s   @r   r   r   	   s9     [112OE0 & &r   N)torch.distributionsr   torch.distributions.gammar   __all__r    r   r   <module>r&      s     + + (&5 &r   