o p i@sddlmZddlZddlmZddlZddlmZddlm Z ddl m Z ddl m Z er:ddlmZdd lmZGd d d e jZddddZdS)) annotationsN) TYPE_CHECKING)check_variable_and_dtype) LayerHelper)exponential_family)in_dynamic_or_pir_mode)Sequence)TensorcseZdZUdZded<dfdd Zeddd Zedd d ZgfdddZ d ddZ d ddZ dddZ ed!ddZ d"ddZZS)# Dirichletaa Dirichlet distribution with parameter "concentration". The Dirichlet distribution is defined over the `(k-1)-simplex` using a positive, length-k vector concentration(`k > 1`). The Dirichlet is identically the Beta distribution when `k = 2`. For independent and identically distributed continuous random variable :math:`\boldsymbol X \in R_k` , and support :math:`\boldsymbol X \in (0,1), ||\boldsymbol X|| = 1` , The probability density function (pdf) is .. math:: f(\boldsymbol X; \boldsymbol \alpha) = \frac{1}{B(\boldsymbol \alpha)} \prod_{i=1}^{k}x_i^{\alpha_i-1} where :math:`\boldsymbol \alpha = {\alpha_1,...,\alpha_k}, k \ge 2` is parameter, the normalizing constant is the multivariate beta function. .. math:: B(\boldsymbol \alpha) = \frac{\prod_{i=1}^{k} \Gamma(\alpha_i)}{\Gamma(\alpha_0)} :math:`\alpha_0=\sum_{i=1}^{k} \alpha_i` is the sum of parameters, :math:`\Gamma(\alpha)` is gamma function. Args: concentration (Tensor): "Concentration" parameter of dirichlet distribution, also called :math:`\alpha`. When it's over one dimension, the last axis denotes the parameter of distribution, ``event_shape=concentration.shape[-1:]`` , axes other than last are consider batch dimensions with ``batch_shape=concentration.shape[:-1]`` . Examples: .. code-block:: python >>> import paddle >>> dirichlet = paddle.distribution.Dirichlet(paddle.to_tensor([1., 2., 3.])) >>> print(dirichlet.entropy()) Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True, -1.24434423) >>> print(dirichlet.prob(paddle.to_tensor([.3, .5, .6]))) Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True, 10.80000019) r concentrationreturnNonecsP|dkst|jdkrtd||_t|jdd|jdddS)Nrz:`concentration` parameter must be at least one dimensional)dimmathprodshape ValueErrorr super__init__)selfr  __class__d/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/paddle/distribution/dirichlet.pyrSs &zDirichlet.__init__cCs|j|jjdddS)zbMean of Dirichlet distribution. Returns: Mean value of distribution. rTZkeepdim)r sumrrrrmean]szDirichlet.meancCs2|jjddd}|j||j|d|dS)zjVariance of Dirichlet distribution. Returns: Variance value of distribution. rTrr)r rpow)rconcentration0rrrvariancefszDirichlet.variancer Sequence[int]cCs,t|tr|nt|}t|j||S)zSample from dirichlet distribution. Args: shape (Sequence[int], optional): Sample shape. Defaults to empty list. ) isinstancetuple _dirichletr expandZ _extend_shape)rrrrrsamplerszDirichlet.samplevaluecCst||S)zProbability density function(PDF) evaluated at value. Args: value (Tensor): Value to be evaluated. Returns: PDF evaluated at value. )paddleexplog_probrr*rrrprob{s zDirichlet.probcCs>t||jddt|jdt|jdS)zoLog of probability density function. Args: value (Tensor): Value to be evaluated. ?r)r+logr rlgammar.rrrr-s zDirichlet.log_probcCsb|jd}|jjd}t|jdt|||t||jdt|jdS)zbEntropy of Dirichlet distribution. Returns: Entropy of distribution. rr0)r rrr+r2Zdigamma)rr"krrrentropys  zDirichlet.entropy tuple[Tensor]cCs|jfSN)r rrrr_natural_parametersszDirichlet._natural_parametersxcCs|dt|dS)Nr)r2rr+)rr8rrr_log_normalizerszDirichlet._log_normalizer)r r r r )r r )rr$r r )r*r r r )r r5)r8r r r )__name__ __module__ __qualname____doc____annotations__rpropertyrr#r)r/r-r4r7r9 __classcell__rrrrr s 0    r r r name str | Noner cCsftr tj|Sd}t|dgd|t|fit}|j|jd}|j |d|id|iid|S)N dirichletr )Zfloat16Zfloat32Zfloat64Zuint16)dtypeAlphaZOut)typeZinputsZoutputsattrs) rr+Z_C_opsrCrrlocalsZ"create_variable_for_type_inferencerDZ append_op)r rAZop_typehelperoutrrrr's( r'r6)r r rArBr r ) __future__rrtypingrr+Zpaddle.base.data_feederrZpaddle.base.layer_helperrZpaddle.distributionrZpaddle.frameworkrcollections.abcrr ZExponentialFamilyr r'rrrrs