/usr/local/lib64/python3.6/site-packages/torch/distributions/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/torch/distributions/__pycache__/laplace.cpython-36.pyc (3442B)
3
Eg @ sL d dl mZ d dlZd dlmZ d dlmZ d dlmZ G dd deZ dS ) )NumberN) constraints) Distribution)
broadcast_allc s e Zd ZdZejejdZejZdZ e
dd Z e
dd Z e
dd Z
d fd d Zd fd
d Zej fddZdd Zdd Zdd Zdd Z ZS )Laplacea
Creates a Laplace distribution parameterized by :attr:`loc` and :attr:`scale`.
Example::
>>> m = Laplace(torch.tensor([0.0]), torch.tensor([1.0]))
>>> m.sample() # Laplace distributed with loc=0, scale=1
tensor([ 0.1046])
Args:
loc (float or Tensor): mean of the distribution
scale (float or Tensor): scale of the distribution
)locscaleTc C s | j S )N)r )self r
G/usr/local/lib64/python3.6/site-packages/torch/distributions/laplace.pymean s z Laplace.meanc C s d| j jd S )N )r pow)r r
r
r variance s zLaplace.variancec C s
d| j S )Nr
g ?g;f?)r )r r
r
r stddev" s zLaplace.stddevNc sR t ||\| _| _t|tr0t|tr0tj }n
| jj }tt | j
||d d S )N)
validate_args) r r r
isinstancer torchSizesizesuperr __init__)r r r r batch_shape) __class__r
r r & s
zLaplace.__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 . s
zLaplace.expandc C s | j |}tj| jj}tjj rptj|| jj| jjdd d }| j| j |j
tj |j j
|jd S | jj|j|jd d}| j| j |j
tj |j S )N)dtypedevicer
)min)Z_extended_shaper finfor r _CZ_get_tracing_stateZrandr r signlog1pabsclampZtinyr Zuniform_Zeps)r Z sample_shapeshaper! ur
r
r rsample7 s
.zLaplace.rsamplec C s8 | j r| j| tjd| j tj|| j | j S )Nr
)r _validate_sampler logr r% r )r valuer
r
r log_probC s
zLaplace.log_probc C sB | j r| j| dd|| j j tj|| j j | j S )Ng ?)r r* r r# r expm1r% r )r r, r
r
r cdfH s
z Laplace.cdfc C s. |d }| j | j|j tjd|j S )Ng ?r
)r r r# r r$ r% )r r, Ztermr
r
r icdfM s z Laplace.icdfc C s dt jd| j S )Nr r
)r r+ r )r r
r
r entropyQ s zLaplace.entropy)N)N)__name__
__module__ __qualname____doc__r realZpositiveZarg_constraintsZsupportZ has_rsamplepropertyr r r r r r r r) r- r/ r1 r2
__classcell__r
r
)r r r s
r )
Znumbersr r Ztorch.distributionsr Z torch.distributions.distributionr Ztorch.distributions.utilsr r r
r
r
r
s