/usr/local/lib64/python3.6/site-packages/torch/distributions/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/torch/distributions/__pycache__/uniform.cpython-36.pyc (3580B)
3
Eg( @ sL d dl mZ d dlZd dlmZ d dlmZ d dlmZ G dd deZ dS ) )NumberN) constraints) Distribution)
broadcast_allc s e Zd ZdZejdddejddddZdZedd Z ed d
Z
ed d Z d fdd Z d fdd Z
ejddddd Zej fddZdd Zdd Zdd Zd d Z ZS ) Uniforma
Generates uniformly distributed random samples from the half-open interval
``[low, high)``.
Example::
>>> m = Uniform(torch.tensor([0.0]), torch.tensor([5.0]))
>>> m.sample() # uniformly distributed in the range [0.0, 5.0)
tensor([ 2.3418])
Args:
low (float or Tensor): lower range (inclusive).
high (float or Tensor): upper range (exclusive).
Fr )Z is_discreteZ event_dim)lowhighTc C s | j | j d S )N )r r )self r G/usr/local/lib64/python3.6/site-packages/torch/distributions/uniform.pymean s z Uniform.meanc C s | j | j d S )N g ?gLXz @)r r )r
r r r stddev! s zUniform.stddevc C s | j | j jdd S )Nr r )r r pow)r
r r r variance% s zUniform.varianceNc sv t ||\| _| _t|tr0t|tr0tj }n
| jj }tt | j
||d | j rrtj | j| jj
rrtdd S )N)
validate_argsz&Uniform is not defined when low>= high)r r r
isinstancer torchSizesizesuperr __init___validate_argsltall
ValueError)r
r r r batch_shape) __class__r r r ) s
zUniform.__init__c sR | j t|}tj|}| jj||_| jj||_tt|j|dd | j |_ |S )NF)r )
Z_get_checked_instancer r r r expandr r r r )r
r Z _instancenew)r r r r 5 s
zUniform.expandc C s t j| j| jS )N)r intervalr r )r
r r r support> s zUniform.supportc C s8 | j |}tj|| jj| jjd}| j|| j| j S )N)dtypedevice)Z_extended_shaper randr r# r$ r )r
Z sample_shapeshaper% r r r rsampleB s
zUniform.rsamplec C sZ | j r| j| | jj|j| j}| jj|j| j}tj|j |tj| j| j S )N)
r _validate_sampler leZtype_asr gtr logmul)r
valueZlbZubr r r log_probG s
zUniform.log_probc C s4 | j r| j| || j | j| j }|jdddS )Nr )minmax)r r( r r clamp)r
r- resultr r r cdfN s
z Uniform.cdfc C s || j | j | j }|S )N)r r )r
r- r3 r r r icdfT s z Uniform.icdfc C s t j| j| j S )N)r r+ r r )r
r r r entropyX s zUniform.entropy)N)N)__name__
__module__ __qualname____doc__r Z dependentZarg_constraintsZ has_rsamplepropertyr
r r r r Zdependent_propertyr" r r r' r. r4 r5 r6
__classcell__r r )r r r s r )
Znumbersr r Ztorch.distributionsr Z torch.distributions.distributionr Ztorch.distributions.utilsr r r r r r
s