Name	Size	Mode	Actions
bernoulli.cpython-36.pyc	4445	0644	editdlrm
beta.cpython-36.pyc	3682	0644	editdlrm
binomial.cpython-36.pyc	4761	0644	editdlrm
categorical.cpython-36.pyc	5813	0644	editdlrm
cauchy.cpython-36.pyc	3310	0644	editdlrm
chi2.cpython-36.pyc	1419	0644	editdlrm
constraints.cpython-36.pyc	22217	0644	editdlrm
constraint_registry.cpython-36.pyc	9895	0644	editdlrm
continuous_bernoulli.cpython-36.pyc	8077	0644	editdlrm
dirichlet.cpython-36.pyc	4085	0644	editdlrm
distribution.cpython-36.pyc	11822	0644	editdlrm
exponential.cpython-36.pyc	3363	0644	editdlrm
exp_family.cpython-36.pyc	2876	0644	editdlrm
fishersnedecor.cpython-36.pyc	3212	0644	editdlrm
gamma.cpython-36.pyc	3455	0644	editdlrm
geometric.cpython-36.pyc	4183	0644	editdlrm
gumbel.cpython-36.pyc	2864	0644	editdlrm
half_cauchy.cpython-36.pyc	2930	0644	editdlrm
half_normal.cpython-36.pyc	2821	0644	editdlrm
independent.cpython-36.pyc	4506	0644	editdlrm
kl.cpython-36.pyc	25844	0644	editdlrm
kumaraswamy.cpython-36.pyc	3040	0644	editdlrm
laplace.cpython-36.pyc	3442	0644	editdlrm
lkj_cholesky.cpython-36.pyc	4575	0644	editdlrm
logistic_normal.cpython-36.pyc	2320	0644	editdlrm
log_normal.cpython-36.pyc	2438	0644	editdlrm
lowrank_multivariate_normal.cpython-36.pyc	8074	0644	editdlrm
mixture_same_family.cpython-36.pyc	7073	0644	editdlrm
multinomial.cpython-36.pyc	5020	0644	editdlrm
multivariate_normal.cpython-36.pyc	8534	0644	editdlrm
negative_binomial.cpython-36.pyc	4149	0644	editdlrm
normal.cpython-36.pyc	4138	0644	editdlrm
one_hot_categorical.cpython-36.pyc	5173	0644	editdlrm
pareto.cpython-36.pyc	2516	0644	editdlrm
poisson.cpython-36.pyc	2682	0644	editdlrm
relaxed_bernoulli.cpython-36.pyc	5595	0644	editdlrm
relaxed_categorical.cpython-36.pyc	5563	0644	editdlrm
studentT.cpython-36.pyc	3483	0644	editdlrm
transformed_distribution.cpython-36.pyc	7335	0644	editdlrm
transforms.cpython-36.pyc	42981	0644	editdlrm
uniform.cpython-36.pyc	3580	0644	editdlrm
utils.cpython-36.pyc	6802	0644	editdlrm
von_mises.cpython-36.pyc	5079	0644	editdlrm
weibull.cpython-36.pyc	2737	0644	editdlrm
__init__.cpython-36.pyc	5981	0644	editdlrm

3 Ego�@shdd
lZddl
mZ
ddlmZ
ddlmZ
ddlmZm Z
ddl mZmZ
dd�Z Gd d �d e�Zd
S)�N)
�constraints)
�Uniform)
�TransformedDistribution)�AffineTransform�PowerTransform)� broadcast_all�euler_constantcCs<d
||}tj
|�
tj
|
�
tj
||
�
}|
tj|�
S)zE Computes nth moment of Kumaraswamy using using torch.lgamma �
)�torch�lgamma�exp)�
a�
b�
nZarg1Z log_value�r�K/usr/local/lib64/python3.6/site-packages/torch/distributions/kumaraswamy.py�_moments s
"
rcsheZ
dZd
Zejejd�ZejZdZ d�f
dd� Z d�f
dd� Zed d ��
Z edd��
Zd d�Z�ZS)�Kumaraswamya Samples from a Kumaraswamy distribution. Example:: >>> m = Kumaraswamy(torch.tensor([1.0]), torch.tensor([1.0])) >>> m.sample() # sample from a Kumaraswamy distribution with concentration alpha=1 and beta=1 tensor([ 0.1729]) 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) )�concentration1�concentration0TNcs�t|
|�\|_
|_tj|jj�
}ttj|jd
�tj|jd�|d�}t|jj �d�
t ddd�t|j
j �d�
g}tt|�j |||d�
dS)Nr
r )
� validate_args)
�exponentg�?)�locZscaleg�)rrrr �finfoZdtyperZ full_liker� reciprocalr�superr�__init__)�selfrrrrZ base_distZ transforms)
� __class__rrr&s








zKumaraswamy.__init__cs<|jt
|�}|jj|
�
|_|jj|
�
|_tt
|�j|
|d
�S)N)
� _instance)Z_get_checked_instancerr�expandrr)rZbatch_shaper�new)
rrrr 1s



zKumaraswamy.expandc

Cst|j
|jd
�S)Nr )rrr)
rrrr�mean7szKumaraswamy.meanc

Cst|j
|jd
�tj|jd
�S)N�)rrrr �powr")
rrrr�variance;szKumaraswamy.variancec
CsTd
|jj
�}
d
|jj
�}tj|jd
�
t}||
|tj|j�
tj|j�
S)Nr )rrrr Zdigammar�log)r�t1�t0ZH0rrr�entropy?s



zKumaraswamy.entropy)
N)
N)�__name__� __module__�__qualname__�__doc__rZpositiveZarg_constraintsZ unit_intervalZsupportZhas_rsamplerr �propertyr"r%r)� __classcell__rr)
rrrs


r)r Ztorch.distributionsrZtorch.distributions.uniformrZ,torch.distributions.transformed_distributionrZtorch.distributions.transformsrrZtorch.distributions.utilsrrrrrrrr�
s