/usr/local/lib64/python3.6/site-packages/torch/distributions/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/torch/distributions/__pycache__/beta.cpython-36.pyc (3682B)
3
EgN
@ s\ d dl mZmZ d dlZd dlmZ d dlmZ d dlm Z d dl
m Z G dd de Z dS ) )RealNumberN) constraints) Dirichlet)ExponentialFamily)
broadcast_allc s e Zd ZdZejejdZejZdZ d fdd Z
d fdd Z e d d
Z
e d d Zf fd
dZdd Zdd Ze dd Ze dd Ze dd Zdd Z ZS ) Betaa8
Beta distribution parameterized by :attr:`concentration1` and :attr:`concentration0`.
Example::
>>> m = Beta(torch.tensor([0.5]), torch.tensor([0.5]))
>>> m.sample() # Beta distributed with concentration concentration1 and concentration0
tensor([ 0.1046])
Args:
concentration1 (float or Tensor): 1st concentration parameter of the distribution
(often referred to as alpha)
concentration0 (float or Tensor): 2nd concentration parameter of the distribution
(often referred to as beta)
)concentration1concentration0TNc st t |tr,t |tr,tjt|t|g}n t||\}}tj||gd}t||d| _t t
| j | jj |d d S )N )
validate_args)
isinstancer torchtensorfloatr stackr
_dirichletsuperr __init__Z _batch_shape)selfr r
r Z concentration1_concentration0) __class__ D/usr/local/lib64/python3.6/site-packages/torch/distributions/beta.pyr s z
Beta.__init__c sD | j t|}tj|}| jj||_tt|j|dd | j|_|S )NF)r ) Z_get_checked_instancer r Sizer expandr r _validate_args)r Z batch_shapeZ _instancenew)r r r r ' s
z Beta.expandc C s | j | j | j S )N)r r
)r r r r mean/ s z Beta.meanc C s* | j | j }| j | j |jd|d S )N r )r r
pow)r totalr r r variance3 s
z
Beta.variancec C s | j j|jddS )Nr r r
)r rsampleselect)r Z sample_shaper r r r# 9 s z Beta.rsamplec C s0 | j r| j| tj|d| gd}| jj|S )Ng ?r r
)r Z_validate_sampler r r log_prob)r valueZ heads_tailsr r r r% < s
z
Beta.log_probc C s
| j j S )N)r entropy)r r r r r' B s z Beta.entropyc C s* | j jd }t|tr"tj|gS |S d S )N.r ).r )r
concentrationr r r r )r resultr r r r E s
zBeta.concentration1c C s* | j jd }t|tr"tj|gS |S d S )N.r ).r )r r( r r r r )r r) r r r r
M s
zBeta.concentration0c C s | j | jfS )N)r r
)r r r r _natural_paramsU s zBeta._natural_paramsc C s" t j|t j| t j|| S )N)r lgamma)r xyr r r _log_normalizerY s zBeta._log_normalizer)N)N)__name__
__module__ __qualname____doc__r ZpositiveZarg_constraintsZ
unit_intervalZsupportZ has_rsampler r propertyr r" r# r% r' r r
r* r.
__classcell__r r )r r r
s r )
Znumbersr r r Ztorch.distributionsr Z torch.distributions.dirichletr Z torch.distributions.exp_familyr Ztorch.distributions.utilsr r r r r r
s