/usr/local/lib64/python3.6/site-packages/torch/distributions/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/torch/distributions/__pycache__/gumbel.cpython-36.pyc (2864B)
3
Eg @ st d dl mZ d dlZd dlZd dlmZ d dlmZ d dlm Z d dl
m Z m Z d dl
mZmZ G dd d e ZdS )
)NumberN) constraints)Uniform)TransformedDistribution)AffineTransform ExpTransform)
broadcast_alleuler_constantc sx e Zd ZdZejejdZejZd fdd Z d fdd Z
dd Z e d
d Z
e d d
Ze dd Zdd Z ZS )Gumbela}
Samples from a Gumbel Distribution.
Examples::
>>> m = Gumbel(torch.tensor([1.0]), torch.tensor([2.0]))
>>> m.sample() # sample from Gumbel distribution with loc=1, scale=2
tensor([ 1.0124])
Args:
loc (float or Tensor): Location parameter of the distribution
scale (float or Tensor): Scale parameter of the distribution
)locscaleNc s t ||\| _| _tj| jj}t|trHt|trHt|j d|j
}n&ttj | j|j tj | jd|j
}t j
tdtj| j dt j
t|| j dg}tt| j|||d d S )N r )r r )
validate_args)r r r torchfinfoZdtype
isinstancer r ZtinyZepsZ full_liker invr Z ones_likesuperr
__init__)selfr r r r Z base_distZ
transforms) __class__ F/usr/local/lib64/python3.6/site-packages/torch/distributions/gumbel.pyr s zGumbel.__init__c s< | j t|}| jj||_| jj||_tt| j||dS )N) _instance)Z_get_checked_instancer
r expandr r )r Z batch_shaper new)r r r r ( s z
Gumbel.expandc C s6 | j r| j| | j| | j }||j | jj S )N)Z_validate_argsZ_validate_sampler r explog)r valueyr r r log_prob/ s
zGumbel.log_probc C s | j | jt S )N)r r r )r r r r mean5 s z Gumbel.meanc C s t jt jd | j S )N )mathpisqrtr )r r r r stddev9 s z
Gumbel.stddevc C s | j jdS )N )r& pow)r r r r variance= s zGumbel.variancec C s | j j dt S )Nr
)r r r )r r r r entropyA s zGumbel.entropy)N)N)__name__
__module__ __qualname____doc__r realZpositiveZarg_constraintsZsupportr r r propertyr! r& r) r*
__classcell__r r )r r r
s
r
)Znumbersr r# r Ztorch.distributionsr Ztorch.distributions.uniformr Z,torch.distributions.transformed_distributionr Z torch.distributions.transformsr r Ztorch.distributions.utilsr r r
r r r r
s