o #iu0@spddlZddlmZmZmZmZddlmZddlZddl m Z ddl m Z ddl mZdgZGdddZdS) N)AnyDictOptionalTuple) deprecated) constraints) lazy_property)_size DistributioncsFeZdZdZdZdZdZededdfddZ e e dfd e j d e j d e effd d Z d@d e j fddZede j fddZede j fddZedeeejffddZede efddZede jfddZede jfddZede jfddZede jfddZe fd e j de jfd!d"Ze fd e j de jfd#d$Ze d%e!d&d'e"de jfd(d)Z#de jde jfd*d+Z$de jde jfd,d-Z%de jde jfd.d/Z&dAd0ede jfd1d2Z'de jfd3d4Z(de jfd5d6Z)e fd e*de+e"d7ffd8d9Z,de jddfd:d;Z-d@dd?Z/Z0S)Br zP Distribution is the abstract base class for probability distributions. FTvaluereturnNcCs|dvrt|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^/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/torch/distributions/distribution.pyset_default_validate_argss  z&Distribution.set_default_validate_args batch_shape event_shape validate_argsc s||_||_|dur ||_|jr{z|j}Wnty-i}t|jdddYnw|D]H\}}t |rsqrtr6rrrstddevs zDistribution.stddev sample_shapecCs6t ||WdS1swYdS)z Generates a sample_shape shaped sample or sample_shape shaped batch of samples if the distribution parameters are batched. N)torchno_gradrsampler,rArrrsamples $zDistribution.samplecCr1)z Generates a sample_shape shaped reparameterized sample or sample_shape shaped batch of reparameterized samples if the distribution parameters are batched. r2rErrrrDr:zDistribution.rsamplez=`sample_n(n)` will be deprecated. Use `sample((n,))` instead.)categoryncCs|t|fS)zq Generates n samples or n batches of samples if the distribution parameters are batched. )rFrBSize)r,rHrrrsample_ns zDistribution.sample_ncCr1)z Returns the log of the probability density/mass function evaluated at `value`. Args: value (Tensor): r2r,r rrrlog_probr8zDistribution.log_probcCr1)z Returns the cumulative density/mass function evaluated at `value`. Args: value (Tensor): r2rKrrrcdfr8zDistribution.cdfcCr1)z Returns the inverse cumulative density/mass function evaluated at `value`. Args: value (Tensor): r2rKrrricdfr8zDistribution.icdfr4cCr1)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. r2)r,r4rrrenumerate_supportszDistribution.enumerate_supportcCr1)z Returns entropy of distribution, batched over batch_shape. Returns: Tensor of shape batch_shape. r2r6rrrentropyszDistribution.entropycCst|S)z Returns perplexity of distribution, batched over batch_shape. Returns: Tensor of shape batch_shape. )rBexprPr6rrr perplexityszDistribution.perplexity.cCs,t|tjs t|}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. )r!rBrIrrrErrr_extended_shapes zDistribution._extended_shapec CsPt|tjs tdt|t|j}||d|jkr.td|d|jd|}|j|j}tt |t |D]\}}|dkr\|dkr\||kr\td|d|dqAz|j }Wnt yyt |jdd d YdSw|dusJ||}|std 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)r!rBTensorr lensizerrzipreversedr9rrrrr$r%r#r&r'r(r)) r,r event_dim_start actual_shapeexpected_shapeijr9r/rrr_validate_samplesX       zDistribution._validate_samplecCsN|durt|j|jkrtd|jjd|jd|dur%|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,clsr3rrr_get_checked_instanceCs z"Distribution._get_checked_instancecsDfddjD}dfdd|D}jjd|dS)Ncsg|] \}}|jvr|qSr)r ).0k_r6rr Lsz)Distribution.__repr__..z, cs@g|]}|dj|dkrj|nj|qS)z: rU)r numelrX)rdpr6rrrgNs2())rrjoinrr&)r, param_names args_stringrr6r__repr__Ks zDistribution.__repr__)N)T)1r& __module__ __qualname____doc__ has_rsamplehas_enumerate_supportr staticmethodboolrrBrIrr+r4propertyrrrstrr Constraintrrr9rVr;r=r>r@rFrDr FutureWarningintrJrLrMrNrOrPrRr rrSr`rcro __classcell__rrr0rr sf'     "  2)rtypingrrrrtyping_extensionsrrBtorch.distributionsrtorch.distributions.utilsr torch.typesr __all__r rrrrs