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 Eg4�@s�dd
lm
Z

ddlmZ
ddlZddljjZddl m Z mZ
ddlm Z
dZdd�Zd d �Zdd�Zddd�
Zdd�Zddd�
ZGdd�d�ZGdd�dee�Zddd�
Zddd�
ZdS) �)
�update_wrapper)
�NumberN)�Dict�Any)
�has_torch_functiong�o��x�?cs�td
d�|D�
�
st
d�
�
tdd�|D�
�
s�ttj�d�
�x*|D]"}
t|
tj�r@t|
j|
jd��Pq@W�f
dd�|D�
}tj |�Stj |�S) a� Given a list of values (possibly containing numbers), returns a list where each value is broadcasted based on the following rules: - `torch.*Tensor` instances are broadcasted as per :ref:`_broadcasting-semantics`. - numbers.Number instances (scalars) are upcast to tensors having the same size and type as the first tensor passed to `values`. If all the values are scalars, then they are upcasted to scalar Tensors. Args: values (list of `numbers.Number`, `torch.*Tensor` or objects implementing __torch_function__) Raises: ValueError: if any of the values is not a `numbers.Number` instance, a `torch.*Tensor` instance, or an instance implementing __torch_function__ c
ss.|]&}
t|
t
j�p$t|
f
�
p$t|
t�V
qdS)
N)� isinstance�torch�Tensorrr)�.0�
v�r�E/usr/local/lib64/python3.6/site-packages/torch/distributions/utils.py� s
z broadcast_all..zqInput arguments must all be instances of numbers.Number, torch.Tensor or objects implementing __torch_function__.c
Ss"g|]}
t|
t
j�pt|
f
�
�qSr)rrr r)r rrrr � sz!broadcast_all..)
�dtype)r�devicec
s4g|],}
t|
t
j�st|
f
�
r"|
nt
j|
f
��
�qSr)rrr rZtensor)r r)
�optionsrr r%s
) �all� ValueError�dictrZget_default_dtyperr rrZbroadcast_tensors)�values�value� new_valuesr)
rr � broadcast_alls










rcCsBtj
j�r.tjtj||
|d
�tj||
|d
��Stj||
|d
�j�S)N)rr)r�_C�_get_tracing_state�normalZzerosZones�emptyZnormal_)�shaperrrrr �_standard_normal+s


rcCs0|
d
kr|S|jd|
�d}|j
|�
jd�
S)z� Sum out ``dim`` many rightmost dimensions of a given tensor. Args: value (Tensor): A tensor of ``.dim()`` at least ``dim``. dim (int): The number of rightmost dims to sum out. r
N�
�����)
r!r!)rZreshape�sum)r�dimZrequired_shaperrr �_sum_rightmost3s


r$FcCs|
rtj
|�
Stj|dd�S)a
Converts a tensor of logits into probabilities. Note that for the binary case, each value denotes log odds, whereas for the multi-dimensional case, the values along the last dimension denote the log probabilities (possibly unnormalized) of the events. r )
r#r!)rZsigmoid�
FZsoftmax)Zlogits� is_binaryrrr �logits_to_probsAs

r'c
Cs tj
|j�
j}
|j|
d
|
d�S)Nr )�min�max)r�finfor�eps�clamp)�probsr+rrr �clamp_probsMs

r.cCs,t|�
}|
r"t
j|�
t
j|�
St
j|�
S)
a$
Converts a tensor of probabilities into logits. For the binary case, this denotes the probability of occurrence of the event indexed by `1`. For the multi-dimensional case, the values along the last dimension denote the probabilities of occurrence of each of the events. )r.r�log�log1p)r-r&Z ps_clampedrrr �probs_to_logitsRs


r1c@s"eZ
dZd
Zdd�Zddd�
ZdS)� lazy_propertyz� Used as a decorator for lazy loading of class attributes. This uses a non-data descriptor that calls the wrapped method to compute the property on first call; thereafter replacing the wrapped method into an instance attribute. cCs|
|_t
||
�
dS)
N)�wrappedr)�selfr3rrr �__init__fs

zlazy_property.__init__NcCsD|
dkrt|j
�
Stj��
|j
|
�
}WdQRXt|
|j
j|�
|S)
N)�_lazy_property_and_propertyr3rZenable_grad�setattr�__name__)r4�instanceZobj_typerrrr �__get__js





zlazy_property.__get__)
N)r8� __module__�__qualname__�__doc__r5r:rrrr r2_s
r2c@seZ
dZd
Zdd�ZdS)r6z�We want lazy properties to look like multiple things. * property when Sphinx autodoc looks * lazy_property when Distribution validate_args looks cCstj
||
�S)
N)�propertyr5)r4r3rrr r5ys
z$_lazy_property_and_property.__init__N)r8r;r<r=r5rrrr r6ss
r6cCs�|jd}t
jj�rJ|
|ks(|
|krJtd|
�d|�d|d
�d��
�
t
j||jd�}||jd d
�|
d
k}|d|f}|S) z� Convert a `D x D` matrix or a batch of matrices into a (batched) vector which comprises of lower triangular elements from the matrix in row order. r zdiag (z) provided is outside [z, z].)
r.r!r!)rrrrr�aranger�view)�mat�diag�
nr?� tril_mask�vecrrr �tril_matrix_to_vec}s

"



rFcCs
d
d|
d
d|
dd|jddt
|
�
|
d
dd}tj|j�
j}tjj�r�t|�
||kr�t d|jd �d�d�
�
t |tj�r�tj|�
j�nt|�
}|j |jd d�tj||f�
�
}tj||jd �}||jdd
�|
d
k}||d|f<|S)z� Convert a vector or a batch of vectors into a batched `D x D` lower triangular matrix containing elements from the vector in row order. r ���g�?zThe size of last dimension is z which cannot be expressed as z3the lower triangular part of a square D x D matrix.N)
r.r!r!r!r!)r�absrr*rr+rr�roundrrr �longZ new_zeros�Sizer?rr@)rErBrCr+rAr?rDrrr �vec_to_tril_matrix�sH




"
"



rN)
F)
F)
r
)
r
)� functoolsrZnumbersrrZtorch.nn.functional�nnZ functionalr%�typingrrZtorch.overridesrZeuler_constantrrr$r'r.r1r2r>r6rFrNrrrr �
s