/usr/local/lib64/python3.6/site-packages/torch/distributions/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/torch/distributions/__pycache__/poisson.cpython-36.pyc (2682B)
3
Eg @ sL d dl mZ d dlZd dlmZ d dlmZ d dlmZ G dd deZ dS ) )NumberN) constraints)ExponentialFamily)
broadcast_allc s e Zd ZdZdejiZejZe dd Z
e dd Z d fdd Z d fd
d Z
ej fd d
Zdd Ze dd Zdd Z ZS )Poissona~
Creates a Poisson distribution parameterized by :attr:`rate`, the rate parameter.
Samples are nonnegative integers, with a pmf given by
.. math::
\mathrm{rate}^k \frac{e^{-\mathrm{rate}}}{k!}
Example::
>>> m = Poisson(torch.tensor([4]))
>>> m.sample()
tensor([ 3.])
Args:
rate (Number, Tensor): the rate parameter
ratec C s | j S )N)r )self r G/usr/local/lib64/python3.6/site-packages/torch/distributions/poisson.pymean s z Poisson.meanc C s | j S )N)r )r r r r
variance" s zPoisson.varianceNc sB t |\| _t|tr tj }n
| jj }tt| j ||d d S )N)
validate_args)
r r
isinstancer torchSizesizesuperr __init__)r r r
batch_shape) __class__r r
r & s
zPoisson.__init__c sD | j t|}tj|}| jj||_tt|j|dd | j|_|S )NF)r
) Z_get_checked_instancer r r r expandr r _validate_args)r r Z _instancenew)r r r
r . s
zPoisson.expandc
C s0 | j |}tj tj| jj|S Q R X d S )N)Z_extended_shaper Zno_gradZpoissonr r )r Z sample_shapeshaper r r
sample6 s
zPoisson.samplec C s: | j r| j| t| j|\}}|j|| |d j S )N )r Z_validate_sampler r Zxlogylgamma)r valuer r r r
log_prob; s
zPoisson.log_probc C s t j| jfS )N)r logr )r r r r
_natural_paramsA s zPoisson._natural_paramsc C s
t j|S )N)r exp)r xr r r
_log_normalizerE s zPoisson._log_normalizer)N)N)__name__
__module__ __qualname____doc__r Z nonnegativeZarg_constraintsZnonnegative_integerZsupportpropertyr 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
s