/usr/local/lib64/python3.6/site-packages/torch/distributions/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/torch/distributions/__pycache__/gamma.cpython-36.pyc (3455B)
3
Eg1 @ sT d dl mZ d dlZd dlmZ d dlmZ d dlmZ dd Z G dd d eZ
dS )
)NumberN) constraints)ExponentialFamily)
broadcast_allc C s
t j| S )N)torch_standard_gamma)
concentration r E/usr/local/lib64/python3.6/site-packages/torch/distributions/gamma.pyr s r c s e Zd ZdZejejdZejZdZdZ e
dd Z e
dd Z d fd
d Z
d fd d
Zej fddZdd Zdd Ze
dd Zdd Z ZS )Gammaa
Creates a Gamma distribution parameterized by shape :attr:`concentration` and :attr:`rate`.
Example::
>>> m = Gamma(torch.tensor([1.0]), torch.tensor([1.0]))
>>> m.sample() # Gamma distributed with concentration=1 and rate=1
tensor([ 0.1046])
Args:
concentration (float or Tensor): shape parameter of the distribution
(often referred to as alpha)
rate (float or Tensor): rate = 1 / scale of the distribution
(often referred to as beta)
)r rateTr c C s | j | j S )N)r r )selfr r r
mean" s z
Gamma.meanc C s | j | jjd S )N )r r pow)r
r r r
variance& s zGamma.varianceNc sR t ||\| _| _t|tr0t|tr0tj }n
| jj }tt | j
||d d S )N)
validate_args) r r r
isinstancer r Sizesizesuperr __init__)r
r r r batch_shape) __class__r r
r * s
zGamma.__init__c sR | j t|}tj|}| jj||_| jj||_tt|j|dd | j |_ |S )NF)r )
Z_get_checked_instancer r r r expandr r r _validate_args)r
r Z _instancenew)r r r
r 2 s
z Gamma.expandc C sD | j |}t| jj|| jj| }|j jtj|j j
d |S )N)min) Z_extended_shaper r r r detachZclamp_r ZfinfodtypeZtiny)r
Z sample_shapeshapevaluer r r
rsample; s
z
Gamma.rsamplec C sd t j|| jj| jjd}| jr(| j| | jt j| j | jd t j| | j| t j | j S )N)r device )
r Z as_tensorr r r# r Z_validate_sampler loglgamma)r
r! r r r
log_probA s
zGamma.log_probc C s4 | j tj| j tj| j d| j tj| j S )Ng ?)r r r% r r& Zdigamma)r
r r r
entropyI s z
Gamma.entropyc C s | j d | j fS )Nr$ )r r )r
r r r
_natural_paramsM s zGamma._natural_paramsc C s& t j|d |d t j|j S )Nr$ )r r& r% Z
reciprocal)r
xyr r r
_log_normalizerQ s zGamma._log_normalizer)N)N)__name__
__module__ __qualname____doc__r ZpositiveZarg_constraintsZsupportZ has_rsampleZ_mean_carrier_measurepropertyr r r r r r r" r' r( r) r,
__classcell__r r )r r
r
s r ) Znumbersr r Ztorch.distributionsr Z torch.distributions.exp_familyr Ztorch.distributions.utilsr r r r r r r
s