U ʹj$0@sdddlZddlmZmZmZmZddlZddlmZddl m Z ddl m Z dgZ GdddZdS)N)AnyDictOptionalTuple) constraints) lazy_property)_size Distributioncs"eZdZdZdZdZdZeeddddZ e e dfe j e j e edfd d Z d=e j d d d Zee j dddZee j dddZeeeejfdddZee edddZee jdddZee jdddZee jdddZee jdddZe fe j e jdd d!Ze fe j e jdd"d#Ze e jd$d%d&Z!e je jdd'd(Z"e je jdd)d*Z#e je jdd+d,Z$d>ee jd-d.d/Z%e jdd0d1Z&e jdd2d3Z'e fe(e)e d4fdd5d6Z*e jddd7d8Z+d?d9d:Z,edd;d<Z-Z.S)@r zP Distribution is the abstract base class for probability distributions. FTN)valuereturncCs|dkr t|t_dS)a Sets whether validation is enabled or disabled. The default behavior mimics Python's ``assert`` statement: validation is on by default, but is disabled if Python is run in optimized mode (via ``python -O``). Validation may be expensive, so you may want to disable it once a model is working. Args: value (bool): Whether to enable validation. )TFN) ValueErrorr _validate_args)r r_/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/distributions/distribution.pyset_default_validate_argss z&Distribution.set_default_validate_args) batch_shape event_shape validate_argsc s||_||_|dk r||_|jrz |j}Wn2tk r\i}t|jdddYnX|D]\}}t |rzqf||j krt t t||trqft ||}||}|sftd|dt|jdt|jdt|dt|d | qftdS) Nz$ does not define `arg_constraints`. zAPlease set `arg_constraints = {}` or initialize the distribution z2with `validate_args=False` to turn off validation.zExpected parameter z ( of shape z) of distribution z to satisfy the constraint , but found invalid values: ) _batch_shape _event_shaper arg_constraintsNotImplementedErrorwarningswarn __class__itemsrZ is_dependent__dict__ isinstancegetattrtypercheckallr __name__tupleshapereprsuper__init__) selfrrrrparam constraintr validrrrr)&s>        :zDistribution.__init__)rcCstdS)a/ Returns a new distribution instance (or populates an existing instance provided by a derived class) with batch dimensions expanded to `batch_shape`. This method calls :class:`~torch.Tensor.expand` on the distribution's parameters. As such, this does not allocate new memory for the expanded distribution instance. Additionally, this does not repeat any args checking or parameter broadcasting in `__init__.py`, when an instance is first created. Args: batch_shape (torch.Size): the desired expanded size. _instance: new instance provided by subclasses that need to override `.expand`. Returns: New distribution instance with batch dimensions expanded to `batch_size`. Nr)r*r _instancerrrexpandMszDistribution.expand)r cCs|jS)zF Returns the shape over which parameters are batched. )rr*rrrrbszDistribution.batch_shapecCs|jS)zJ Returns the shape of a single sample (without batching). )rr2rrrriszDistribution.event_shapecCstdS)a  Returns a dictionary from argument names to :class:`~torch.distributions.constraints.Constraint` objects that should be satisfied by each argument of this distribution. Args that are not tensors need not appear in this dict. Nr/r2rrrrpszDistribution.arg_constraintscCstdS)z Returns a :class:`~torch.distributions.constraints.Constraint` object representing this distribution's support. Nr/r2rrrsupportzszDistribution.supportcCstdS)z7 Returns the mean of the distribution. Nr/r2rrrmeanszDistribution.meancCst|jddS)z7 Returns the mode of the distribution. z does not implement modeN)rrr2rrrmodeszDistribution.modecCstdS)z; Returns the variance of the distribution. Nr/r2rrrvarianceszDistribution.variancecCs |jS)zE Returns the standard deviation of the distribution. )r6sqrtr2rrrstddevszDistribution.stddev) sample_shaper c Cs*t||W5QRSQRXdS)z Generates a sample_shape shaped sample or sample_shape shaped batch of samples if the distribution parameters are batched. N)torchZno_gradrsampler*r9rrrsamples zDistribution.samplecCstdS)z Generates a sample_shape shaped reparameterized sample or sample_shape shaped batch of reparameterized samples if the distribution parameters are batched. Nr/r<rrrr;szDistribution.rsample)nr cCstdt|t|fS)zq Generates n samples or n batches of samples if the distribution parameters are batched. z6sample_n will be deprecated. Use .sample((n,)) instead)rr UserWarningr=r:Size)r*r>rrrsample_ns zDistribution.sample_ncCstdS)z Returns the log of the probability density/mass function evaluated at `value`. Args: value (Tensor): Nr/r*r rrrlog_probszDistribution.log_probcCstdS)z Returns the cumulative density/mass function evaluated at `value`. Args: value (Tensor): Nr/rBrrrcdfszDistribution.cdfcCstdS)z Returns the inverse cumulative density/mass function evaluated at `value`. Args: value (Tensor): Nr/rBrrricdfszDistribution.icdf)r1r cCstdS)ar Returns tensor containing all values supported by a discrete distribution. The result will enumerate over dimension 0, so the shape of the result will be `(cardinality,) + batch_shape + event_shape` (where `event_shape = ()` for univariate distributions). Note that this enumerates over all batched tensors in lock-step `[[0, 0], [1, 1], ...]`. With `expand=False`, enumeration happens along dim 0, but with the remaining batch dimensions being singleton dimensions, `[[0], [1], ..`. To iterate over the full Cartesian product use `itertools.product(m.enumerate_support())`. Args: expand (bool): whether to expand the support over the batch dims to match the distribution's `batch_shape`. Returns: Tensor iterating over dimension 0. Nr/)r*r1rrrenumerate_supportszDistribution.enumerate_supportcCstdS)z Returns entropy of distribution, batched over batch_shape. Returns: Tensor of shape batch_shape. Nr/r2rrrentropyszDistribution.entropycCst|S)z Returns perplexity of distribution, batched over batch_shape. Returns: Tensor of shape batch_shape. )r:exprGr2rrr perplexityszDistribution.perplexity.cCs,t|tjst|}t||j|jS)ax Returns the size of the sample returned by the distribution, given a `sample_shape`. Note, that the batch and event shapes of a distribution instance are fixed at the time of construction. If this is empty, the returned shape is upcast to (1,). Args: sample_shape (torch.Size): the size of the sample to be drawn. )rr:r@rrr<rrr_extended_shapes  zDistribution._extended_shapec CsVt|tjstdt|t|j}||d|jkr\td|d|jd|}|j|j}tt |t |D]6\}}|dkr|dkr||krtd|d|dqz |j }Wn0t k rt |jdd d YdSX|dk st||}|sRtd t|jd t|jd t|dt|d| dS)a Argument validation for distribution methods such as `log_prob`, `cdf` and `icdf`. The rightmost dimensions of a value to be scored via these methods must agree with the distribution's batch and event shapes. Args: value (Tensor): the tensor whose log probability is to be computed by the `log_prob` method. Raises ValueError: when the rightmost dimensions of `value` do not match the distribution's batch and event shapes. z/The value argument to log_prob must be a TensorNz5The right-most size of value must match event_shape: z vs .z9Value is not broadcastable with batch_shape+event_shape: z% does not define `support` to enable z;sample validation. Please initialize the distribution with z-`validate_args=False` to turn off validation.zExpected value argument (rz) to be within the support (z) of the distribution r)rr:Tensorr lensizerrzipreversedr3rrrrAssertionErrorr"r#r!r$r%r&r') r*r Zevent_dim_startZ actual_shapeZexpected_shapeijr3r-rrr_validate_samples>      4zDistribution._validate_samplecCsN|dkr4t|j|jkr4td|jjd|jd|dkrJ|t|S|S)Nz Subclass z of zR that defines a custom __init__ method must also define a custom .expand() method.)r!r)rrr$__new__)r*clsr0rrr_get_checked_instance@s z"Distribution._get_checked_instancecsDfddjD}dfdd|D}jjd|dS)Ncsg|]\}}|jkr|qSr)r).0k_r2rr Is z)Distribution.__repr__..z, cs@g|]8}|dj|dkr*j|n j|qS)z: rL)rZnumelrO)rYpr2rrr\Ks())rrjoinrr$)r*Z param_names args_stringrr2r__repr__Hs zDistribution.__repr__)N)T)N)/r$ __module__ __qualname____doc__Z has_rsampleZhas_enumerate_supportr staticmethodboolrr:r@rr)r1propertyrrrstrr Constraintrrr3rMr4r5r6r8r=r;intrArCrDrErFrGrIrrrJrUrXrb __classcell__rrr.rr sX'       2 )rtypingrrrrr:Ztorch.distributionsrZtorch.distributions.utilsrZ torch.typesr__all__r rrrrs