/usr/local/lib64/python3.6/site-packages/torch/distributions/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/torch/distributions/__pycache__/chi2.cpython-36.pyc (1419B)
3
Eg @ s, d dl mZ d dlmZ G dd deZdS ) )constraints)Gammac sF e Zd ZdZdejiZd
fdd Zd fdd Ze dd Z
ZS )Chi2ar
Creates a Chi2 distribution parameterized by shape parameter :attr:`df`.
This is exactly equivalent to ``Gamma(alpha=0.5*df, beta=0.5)``
Example::
>>> 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
dfNc s t t| jd| d|d d S )Ng ?)
validate_args)superr __init__)selfr r ) __class__ D/usr/local/lib64/python3.6/site-packages/torch/distributions/chi2.pyr s z
Chi2.__init__c s | j t|}tt| j||S )N)Z_get_checked_instancer r expand)r Zbatch_shapeZ _instancenew)r
r r r
s zChi2.expandc C s
| j d S )N )Z
concentration)r r r r r s zChi2.df)N)N)__name__
__module____qualname____doc__r ZpositiveZarg_constraintsr r
propertyr
__classcell__r r )r
r r s
r N)Ztorch.distributionsr Ztorch.distributions.gammar r r r r r
s