/usr/local/lib64/python3.6/site-packages/torch/distributions/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/torch/distributions/__pycache__/studentT.cpython-36.pyc (3483B)
3
Eg
@ s` d dl Z d dlZd dlmZmZ d dlmZmZ d dlm Z d dl
m Z m Z G dd de Z
dS ) N)infnan)Chi2 constraints) Distribution)_standard_normal
broadcast_allc s e Zd ZdZejejejdZejZdZ e
dd Z e
dd Z d fd d Z
d fd
d Zej fddZdd Zdd Z ZS )StudentTa
Creates a Student's t-distribution parameterized by degree of
freedom :attr:`df`, mean :attr:`loc` and scale :attr:`scale`.
Example::
>>> m = StudentT(torch.tensor([2.0]))
>>> m.sample() # Student's t-distributed with degrees of freedom=2
tensor([ 0.1046])
Args:
df (float or Tensor): degrees of freedom
loc (float or Tensor): mean of the distribution
scale (float or Tensor): scale of the distribution
)dflocscaleTc C s" | j jtjd}t|| jdk< |S )N)
memory_format )r clonetorchcontiguous_formatr r
)selfm r H/usr/local/lib64/python3.6/site-packages/torch/distributions/studentT.pymean s z
StudentT.meanc C s~ | j jtjd}| j| j dk jd| j | j dk | j | j dk d || j dk< t|| j dk| j dk@ < t|| j dk< |S )N)r
r )r
r r r r powr r )r r r r r variance$ s
DzStudentT.variance ?Nc sF t |||\| _| _| _t| j| _| jj }tt| j ||d d S )N)
validate_args)
r r
r r r _chi2sizesuperr __init__)r r
r r r batch_shape) __class__r r r , s
zStudentT.__init__c sn | j t|}tj|}| jj||_| jj||_| jj||_| jj||_t t|j
|dd | j |_ |S )NF)r ) Z_get_checked_instancer r Sizer
expandr r r r r _validate_args)r r! Z _instancenew)r" r r r$ 2 s
zStudentT.expandc C sP | j |}t|| jj| jjd}| jj|}|tj|| j }| j | j
| S )N)dtypedevice) Z_extended_shaper r
r' r( r rsampler Zrsqrtr r )r Z sample_shapeshapeXZYr r r r) = s
zStudentT.rsamplec C s | j r| j| || j | j }| jj d| jj dtjtj tj d| j tj d| jd }d| jd tj
|d | j | S )Ng ?g ?g @g ) r% Z_validate_sampler r logr
mathpir lgammalog1p)r valueyr, r r r log_probK s
6zStudentT.log_probc C s| t jd| j tjd t jd| jd }| jj d| jd t jd| jd t jd| j d| jj | S )Ng ?r )r r1 r
r/ r r. Zdigamma)r Zlbetar r r entropyV s .zStudentT.entropy)r r N)N)__name__
__module__ __qualname____doc__r ZpositiverealZarg_constraintsZsupportZ has_rsamplepropertyr r r r$ r r# r) r5 r6
__classcell__r r )r" r r
s r )r/ r Z
torch._sixr r Ztorch.distributionsr r Z torch.distributions.distributionr Ztorch.distributions.utilsr r r r r r r
s