o p i!.@sddlmZddlZddlmZddlZddlZddlmZddl m Z m Z ddl m Z ddlmZmZerPddlmZdd lmZdd lmZdd lmZmZGd d d ZdS)) annotationsN) TYPE_CHECKING)_C_ops)check_variable_and_dtype convert_dtype)Variable)in_dynamic_or_pir_mode in_pir_mode)Sequence) TypeGuard)Tensor)NestedNumericSequence TensorLikecseZdZdZ d?d@fdd ZedAd d ZedAd d ZedBddZedBddZ gfdCddZ gfdCddZ dBddZ dDddZ dEddZdEd d!ZdEd"d#ZdFd&d'ZdGd+d,ZdHd.d/ZdId1d2Z 3dJdKd8d9Z 3dJdLd;d<ZdHd=d>ZZS)M Distributiona> The abstract base class for probability distributions. Functions are implemented in specific distributions. Args: batch_shape(Sequence[int], optional): independent, not identically distributed draws, aka a "collection" or "bunch" of distributions. event_shape(Sequence[int], optional): the shape of a single draw from the distribution; it may be dependent across dimensions. For scalar distributions, the event shape is []. For n-dimension multivariate distribution, the event shape is [n]. batch_shape Sequence[int] event_shapereturnNonecs>t|tr|nt||_t|tr|nt||_tdSN) isinstancetuple _batch_shape _event_shapesuper__init__)selfrr __class__rg/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/paddle/distribution/distribution.pyr6szDistribution.__init__cC|jS)zeReturns batch shape of distribution Returns: Sequence[int]: batch shape )rrrrr rFzDistribution.batch_shapecCr!)zeReturns event shape of distribution Returns: Sequence[int]: event shape )rr"rrr rOr#zDistribution.event_shaper cCt)zMean of distributionNotImplementedErrorr"rrr meanXzDistribution.meancCr$)zVariance of distributionr%r"rrr variance]r(zDistribution.varianceshapecCr$)zSampling from the distribution.r%rr*rrr samplebzDistribution.samplecCr$)zreparameterized sampler%r+rrr rsamplefr-zDistribution.rsamplecCr$)z The entropy of the distribution.r%r"rrr entropyjr-zDistribution.entropyothercCr$)z7The KL-divergence between self distributions and other.r%)rr0rrr kl_divergencenr-zDistribution.kl_divergencevaluecCs||S)zProbability density/mass function evaluated at value. Args: value (Tensor): value which will be evaluated )log_probexprr2rrr probrszDistribution.probcCr$)z&Log probability density/mass function.r%r5rrr r3zr-zDistribution.log_probcCr$)zProbability density/mass function. Note: This method will be deprecated in the future, please use `prob` instead. r%r5rrr probs~szDistribution.probs sample_shapeSequence[int] | TensorcCst|t|jt|jS)zcompute shape of the sample Args: sample_shape (Sequence[int]|Tensor): sample shape Returns: Tensor: generated sample data shape )rrr)rr8rrr _extend_shapes  zDistribution._extend_shapeargs"TensorLike | NestedNumericSequenceTypeGuard[Tensor]cGsBd}d}|D]}t|ttjjfrd}qd}q|r|rtd|S)z Argument validation for distribution args Args: value (float, list, numpy.ndarray, Tensor) Raises ValueError: if one argument is Tensor, all arguments should be Tensor FTz9if one argument is Tensor, all arguments should be Tensor)rrpaddlepirValue ValueError)rr;Z is_variableZ is_numberargrrr _validate_argss zDistribution._validate_argstuple[Tensor, ...]c Gs&g}g}d}|D]M}t|ttttjttjj fs!t dt |t|tjj r.| |qt |}|j}t|dkrLt|dkrGtd|d}||}| |q|j}|D]3}t|tjj rj| |q[t||\} } tr|t| j} ntjj|d} t| | | | q[t|S)z Argument convert args to Tensor Args: value (float, list, numpy.ndarray, Tensor) Returns: Tensor of args. gZType of input args must be float, list, tuple, numpy.ndarray or Tensor, but received type float32float64zadata type of argument only support float32 and float64, your argument will be convert to float32.dtype)rfloatlistrnpndarrayrr>r?r@ TypeErrortypeappendarrayrIstrwarningswarnastypeZbroadcast_arraysr Zzerosr*ZtensorZ create_tensorZassign) rr;Z numpy_argsZ variable_argstmprBZarg_npZ arg_dtyperIZarg_broadcasted_Z arg_variablerrr _to_tensorsH           zDistribution._to_tensorparamcCst|r ||jStr5trt|dddgd|j|jkr3t|jdvr3t dt ||jS|St|dddgd|j|jkrQt dtj ||jdS|S)a Log_prob and probs methods have input ``value``, if value's dtype is different from param, convert value's dtype to be consistent with param's dtype. Args: param (Tensor): low and high in Uniform class, loc and scale in Normal class. value (Tensor): The input tensor. Returns: value (Tensor): Change value's dtype if value's dtype is different from param. r2rFrGr3)rFrGztdtype of input 'value' needs to be the same as parameters of distribution class. dtype of 'value' will be converted.rH) r>Z is_complexrUrIrr rrrSrTrcast)rrYr2rrr _check_values_dtype_in_probss2    z)Distribution._check_values_dtype_in_probsFr7float | Tensor is_binaryboolcCs$|r t|t| St|S)a Converts probabilities into logits. For the binary, probs denotes the probability of occurrence of the event indexed by `1`. For the multi-dimensional, values of last axis denote the probabilities of occurrence of each of the events. )r>loglog1p)rr7r]rrr _probs_to_logitss  zDistribution._probs_to_logitslogitscCs$|r tjj|Stjjj|ddS)z Converts logits into probabilities. For the binary, each value denotes log odds, whereas for the multi-dimensional case, the values along the last dimension denote the log probabilities of the events. )Zaxis)r>nnZ functionalZsigmoidZsoftmax)rrbr]rrr _logits_to_probs(s  zDistribution._logits_to_probsc s|D]}t|ttttjttjj fst dt |qt dd|DsJt |D]}t|ttjj fr;|jnq+fdd|D}t|St|S)a Returns a list where each arg is broadcasted. Scalar args are upcast to tensors having the same data type as the first Tensor passed to `args`. If all the args are scalars, then they are upcasted to Tensors with paddle default data type. Args: value (float, list, numpy.ndarray, Tensor) Returns: Broadcasted Tensor of args. rEcss"|] }t|ttjjfVqdSr)rrr>r?r@.0rBrrr Ls z.Distribution._broadcast_all..cs0g|]}t|ttjjfr|ntj|dqS)rH)rrr>r?r@Z to_tensorrfrHrr Ts  z/Distribution._broadcast_all..)rrJrKrrLrMrr>r?r@rNrOallZget_default_dtyperIZbroadcast_tensors)rr;rBnew_argsrrHr _broadcast_all6s.    zDistribution._broadcast_all)rr)rrrrrr)rr)rr )r*rrr )r0rrr )r2r rr )r8r9rr )r;r<rr=)r;r<rrD)rYr r2r rr )F)r7r\r]r^rr )rbr\r]r^rr )__name__ __module__ __qualname____doc__rpropertyrrr'r)r,r.r/r1r6r3r7r:rCrXr[rarerl __classcell__rrrr r(s8            ;/  r) __future__rrStypingrnumpyrLr>rZpaddle.base.data_feederrrZpaddle.base.frameworkrZpaddle.frameworkrr collections.abcr Ztyping_extensionsr r Zpaddle._typingr rrrrrr s