tj1rddlZddlmZddlZddlmZddlmZddlmZddl m Z dgZ GddZ dS) N) deprecated)Tensor) constraints) lazy_property)_size Distributionc *eZdZdZdZdZdZededdfdZ e j e j dfde j d e j d edzddffd Z d'de fd Zede j fd Zede j fdZedeeejffdZedejdzfdZedefdZedefdZedefdZedefdZe j fde defdZe j fde defdZede de!defdZ"dedefdZ#dedefdZ$dedefdZ%d(dedefd Z&defd!Z'defd"Z(e j fde de j fd#Z)deddfd$Z*d'd%Z+defd&Z,xZ-S))raS Distribution is the abstract base class for probability distributions. Args: batch_shape (torch.Size): The shape over which parameters are batched. event_shape (torch.Size): The shape of a single sample (without batching). validate_args (bool, optional): Whether to validate arguments. Default: None. FTvaluereturnNc4|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 s ^/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/torch/distributions/distribution.pyset_default_validate_argsz&Distribution.set_default_validate_argss#  % % &+ ### batch_shape event_shape validate_argsc||_||_|||_|jr@ |j}n6#t$r)i}t j|jddzdzdYnwxYw|D]\}}tj |r||j vr1ttt||trTt||}||}t#j|s_t'd|dt|jdt+|jd t/|d t/|d | t1dS) Nz$ does not define `arg_constraints`. zAPlease set `arg_constraints = {}` or initialize the distribution z2with `validate_args=False` to turn off validation. stacklevelzExpected parameter z ( of shape z) of distribution z to satisfy the constraint , but found invalid values: ) _batch_shape _event_shaperarg_constraintsNotImplementedErrorwarningswarn __class__itemsr is_dependent__dict__ isinstancegetattrtyperchecktorch _is_all_truer __name__tupleshapereprsuper__init__) selfrrrrparam constraintr validr!s rr0zDistribution.__init__.s ('  $"/D     "&"6&   "$ ~KKKYZJK !  &5%:%:%<%<  !z+J77 --*DJJ.. 33-e,,"((//)%00$>e>> KK0>>>+/::>>6:*5E5E>>7< >> s*0AAct)a0 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_shape`. r)r1r _instances rexpandzDistribution.expandVs &"!rc|jS)zF Returns the shape over which parameters are batched. )rr1s rrzDistribution.batch_shapek   rc|jS)zJ Returns the shape of a single sample (without batching). )rr:s rrzDistribution.event_shaperr;rct)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. r6r:s rrzDistribution.arg_constraintsy "!rct)z Returns a :class:`~torch.distributions.constraints.Constraint` object representing this distribution's support. r6r:s rsupportzDistribution.support "!rct)z7 Returns the mean of the distribution. r6r:s rmeanzDistribution.mean "!rc0t|jd)z7 Returns the mode of the distribution. z does not implement mode)rr!r:s rmodezDistribution.modes "T^"M"M"MNNNrct)z; Returns the variance of the distribution. r6r:s rvariancezDistribution.variancerDrc4|jS)zE Returns the standard deviation of the distribution. )rHsqrtr:s rstddevzDistribution.stddevs }!!###r sample_shapectj5||cdddS#1swxYwYdS)z Generates a sample_shape shaped sample or sample_shape shaped batch of samples if the distribution parameters are batched. N)r)no_gradrsampler1rLs rsamplezDistribution.samples ]__ . .<< -- . . . . . . . . . . . . . . . . . .s 6::ct)z Generates a sample_shape shaped reparameterized sample or sample_shape shaped batch of reparameterized samples if the distribution parameters are batched. r6rPs rrOzDistribution.rsamplerArz=`sample_n(n)` will be deprecated. Use `sample((n,))` instead.)categoryncR|tj|fS)zq Generates n samples or n batches of samples if the distribution parameters are batched. )rQr)Size)r1rTs rsample_nzDistribution.sample_ns"{{5:qd++,,,rct)z Returns the log of the probability density/mass function evaluated at `value`. Args: value (Tensor): r6r1r s rlog_probzDistribution.log_probr>rct)z Returns the cumulative density/mass function evaluated at `value`. Args: value (Tensor): r6rYs rcdfzDistribution.cdfr>rct)z Returns the inverse cumulative density/mass function evaluated at `value`. Args: value (Tensor): r6rYs ricdfzDistribution.icdfr>rr8ct)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. r6)r1r8s renumerate_supportzDistribution.enumerate_supports ,"!rct)z Returns entropy of distribution, batched over batch_shape. Returns: Tensor of shape batch_shape. r6r:s rentropyzDistribution.entropys "!rcNtj|S)z Returns perplexity of distribution, batched over batch_shape. Returns: Tensor of shape batch_shape. )r)exprbr:s r perplexityzDistribution.perplexitysy(((rct|tjstj|}tj||jz|jzS)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%r)rVrrrPs r_extended_shapezDistribution._extended_shape sG, 33 4 :l33Lz,)::T=NNOOOrc t|tjstdt |t |jz }||d|jkr-td|d|jd|}|j|jz}tt|t|D]-\}}|dkr"|dkr||krtd|d|d. |j }n5#t$r(tj |jdd zd zd YdSwxYw|td ||}tj|s\tdt%|jdt)|jdt-|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.rrzsupport is unexpectedly NonezExpected value argument (rz) to be within the support (z) of the distribution r)r%r)rr lensizerrzipreversedr@rrr r!AssertionErrorr(r*r'r+r,r-r.) r1r event_dim_start actual_shapeexpected_shapeijr@r4s r_validate_samplezDistribution._validate_samplesa%.. PNOO Oejjll++c$2C.D.DD ::<<(( )T-> > >n nnZ^Zknnn zz|| *T->>..0H0HII  DAqAvv!q&&Q!VV sP\ssbpsss lGG"    M>HHHOPAB      FF  ? !?@@ @ e$$!%(( 6KK(6649%+4F4F66-1']]66(,Dzz66/4 66   s.D66.E('E(c|Bt|j|jkr%td|jjd|jd|"|t|n|S)Nz Subclass z of zR that defines a custom __init__ method must also define a custom .expand() method.)r'r0rr!r+__new__)r1clsr7s r_get_checked_instancez"Distribution._get_checked_instanceLs|  d!4 !D!D%>DN3>>>>> ,5+z)Distribution.__repr__..Us%MMMQ!t}:L:Lq:L:L:Lrz, cg|]V}|dj|dkr j|nj|WS)z: rj)r$numelrl)r}pr1s rrz)Distribution.__repr__..Wst   hhDM!,<,B,B,D,D,I,I a((t}]^O_OdOdOfOfhh   r())rjoinr!r+)r1 param_names args_strings` r__repr__zDistribution.__repr__TstMMMM$"6MMM ii    $     ~&,{:S@@r)N)T).r+ __module__ __qualname____doc__ has_rsamplehas_enumerate_supportr staticmethodboolrr)rVr0rr8propertyrrdictstrr Constraintrr@rrCrFrHrKrQrOr FutureWarningintrWrZr\r^r`rbrergruryr __classcell__)r!s@rrrsYK!N,,$,,,\,$#-%*,,",%*,,%) &&Z&Z&d{ &  &&&&&&P""%""""*!UZ!!!X! !UZ!!!X! "c;+A&A!B"""X""/$6"""X""f"""X" OfOOOXO "&"""X" $$$$X$ ,65:<<..5.F....-7EJLL""E"V""""ZG-#-&--- -"f"""""""F"""""&"V"""""""""""0""""")F))))5?EJLL P PE PUZ P P P P2f22222hLLLLA#AAAAAAAAr) rtyping_extensionsrr)rtorch.distributionsrtorch.distributions.utilsr torch.typesr__all__rr|rrrs(((((( ++++++333333  MAMAMAMAMAMAMAMAMAMAr