o + i#@sTddlmZddlZddlmZmZddlmZddlm Z ddl m Z dd l m Z mZdd lmZmZdd lmZdd lmZdd lmZmZ  d@dAddZ  d@dAddZ   dBdCd!d"Z   dBdCd#d$Z % dDdEd'd(Z % dDdFd+d,Z %  - .dGdHd/d0ZdId2d3Z dJd4d5Z!dJd6d7Z" dKdLd:d;Z#dJdd?Z%dS)N) annotationsN)in_dygraph_mode in_pir_modeConstantDirac)calculate_gain) KaimingNormalKaimingUniform)NormalTruncatedNormal OrthogonalUniform) XavierNormal XavierUniformfan_in leaky_relutensor paddle.Tensorafloatmodestr nonlinearityreturnpaddle.Tensor | NonecCt|||d}||S)aModify tensor inplace using Kaiming uniform method. Args: tensor (Tensor): Paddle Tensor. a (float, optional): The negative slope of the rectifier used after this layer. Defaults to 0. mode (str, optional): Mode to compute the fan. Choose from ["fan_in", "fan_out"]. When set to 'fan_in', the fan_in parameter is used for initialization. When set to 'fan_out', the out_features of trainable Tensor will be used. Default is 'fan_in'. nonlinearity (str, optional): Nonlinearity method name. Defaults to "leaky_relu". Returns: Tensor: Initialized tensor. Znegative_sloperr)r rrrrinitr%Z/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/paddle/nn/init.pykaiming_uniform_sr'cCr!)aModify tensor inplace using Kaiming normal method. Args: tensor (Tensor): Paddle Tensor. a (float, optional): The negative slope of the rectifier used after this layer. Defaults to 0. mode (str, optional): Mode to compute the fan. Choose from ["fan_in", "fan_out"]. When set to 'fan_in', the fan_in parameter is used for initialization. When set to 'fan_out', the out_features of trainable Tensor will be used. Default is 'fan_in'. nonlinearity (str, optional): Nonlinearity method name. Defaults to "leaky_relu". Returns: Tensor: Initialized tensor. r")r r#r%r%r&kaiming_normal_:sr(?gain float | Nonefan_outcCt|||d}||S)aModify tensor inplace using Xavier uniform method. Args: tensor (Tensor): Paddle Tensor. gain (float, optional): Scaling Tensor. Default is 1.0. fan_in (float|None, optional): fan_in for Xavier initialization, which is inferred from the Tensor. Default is None. fan_out (float|None, optional): fan_out for Xavier initialization, which is inferred from the Tensor. Default is None. Returns: Tensor: Initialized tensor. r*rr,)rrr*rr,r$r%r%r&xavier_uniform_T r0cCr-)aModify tensor inplace using Xavier normal method. Args: tensor (Tensor): Paddle Tensor. gain (float, optional): Scaling Tensor. Default is 1.0. fan_in (float|None, optional): fan_in for Xavier initialization, which is inferred from the Tensor. Default is None. fan_out (float|None, optional): fan_out for Xavier initialization, which is inferred from the Tensor. Default is None. Returns: Tensor: Initialized tensor. r.)rr/r%r%r&xavier_normal_pr1r2bcCt||d}||S)a^Modify tensor inplace using uniform method. Args: tensor (Tensor): Paddle Tensor. low (float, optional): Lower boundary of the uniform distribution. Default is :math:`-1.0`. high (float, optional): Upper boundary of the uniform distribution. Default is :math:`1.0`. Returns: Tensor: Initialized tensor. )lowhighr)rrr4r$r%r%r&uniform_ r8meanstdcCr5)aLModify tensor inplace using normal method. Args: tensor (Tensor): Paddle Tensor. mean (float|complex, optional): mean of the normal distribution. Default is 0.0. std (float, optional): standard deviation of the normal distribution. Default is 1.0. Returns: Tensor: Initialized tensor. )r:r;)r)rr:r;r$r%r%r&normal_r9r<@cCst||||d}||S)aModify tensor inplace using truncated normal method. Args: tensor (Tensor): Paddle Tensor. mean (float|complex, optional): mean of the normal distribution. Default is 0.0. std (float, optional): standard deviation of the normal distribution. Default is 1.0. a (float, optional): The minimum cutoff value. Default is -2.0. b (float, optional): The maximum cutoff value. Default is 2.0. Returns: Tensor: Initialized tensor. )r:r;rr4)r)rr:r;rr4r$r%r%r& trunc_normal_sr?valcCt|d}||S)zModify tensor inplace using constant method. Args: tensor (Tensor): Paddle Tensor. value (float32|float64, optional): constant value to initialize the parameter. Returns: Tensor: Initialized tensor. valuer)rr@r$r%r%r& constant_ rDcCtdd}||S)zFill the input Tensor with the scalar value 1. Args: tensor (Tensor): Paddle Tensor. Returns: Tensor: Initialized tensor. r)rBrrr$r%r%r&ones_ rHcCrF)zFill the input Tensor with the scalar value 0. Args: tensor (Tensor): Paddle Tensor. Returns: Tensor: Initialized tensor. r3rBrrGr%r%r&zeros_rIrJgroupsintcCrA)a4Initialize the 3D/4D/5D Tensor with Dirac delta function. Args: tensor (Tensor): Paddle Tensor. groups (int|None, optional): 0-dimension of the Tensor will be divided by groups, each group has the same value. Default: 1. Returns: Tensor: Initialized tensor. )rKr )rrKr$r%r%r&dirac_rErMcCst|jdkrtdt|jdtr+tj|jd|jd|jd}||dStr?tj|jd|jd|jd}|St d) zFill the 2-dimensional input Tensor with the identity matrix. Args: tensor (Tensor): Paddle Tensor. Returns: Tensor: Initialized tensor. rz+Only support 2 dimensional tensor, but got .rr)dtypeNz-Only support run in dygraph mode or PIR mode.) lenshapeAssertionErrorrpaddleeyerOZ_share_underline_tensor_torNotImplementedError)rZ new_tensorr%r%r&eye_s$  rVcCrA)aFill the input Tensor with a (semi) orthogonal matrix. Args: tensor (Tensor): Paddle Tensor. gain(float, optional): The multiplication coefficient for initialized tensor. Default: 1.0. Returns: Tensor: Initialized tensor. )r*r)rr*r$r%r%r& orthogonal_1s rW)rrr) rrrrrrrrrr )r)NN) rrr*rrr+r,r+rr )r3r))rrrrr4rrr )rrr:rr;rrr )r3r)r=r>) rrr:rr;rrrr4rrr )rrr@rrr )rrrr )r)rrrKrLrr )rrr*rrr )& __future__rrSZbase.frameworkrrZinitializer.constantrZinitializer.diracr Zinitializer.initializerr Zinitializer.kaimingr r Zinitializer.normalrrZinitializer.orthogonalrZinitializer.uniformrZinitializer.xavierrrr'r(r0r2r8r<r?rDrHrJrMrVrWr%r%r%r&s\                #