o
    ûpi½  ã                   @   s   d dl mZmZmZmZmZ d dlmZmZm	Z	 d dl
mZmZ d dlmZ d dlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z, d dl-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZD d dlEmFZFmGZGmHZHmIZImJZJ d dlKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZT d d	lUmVZV d d
lWmXZX d dlYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZompZp d dlqmrZrmsZsmtZtmuZumvZvmwZwmxZxmyZymzZzm{Z{m|Z|m}Z} d dl~mZm€Z€mZm‚Z‚mƒZƒm„Z„m…Z…m†Z†m‡Z‡mˆZˆm‰Z‰mŠZŠm‹Z‹mŒZŒmZmŽZŽmZmZm‘Z‘ d dl’m“Z“m”Z”m•Z•m–Z–m—Z—m˜Z˜m™Z™mšZšm›Z› d dlœmZmžZžmŸZŸm Z m¡Z¡m¢Z¢ d dl£m¤Z¤m¥Z¥m¦Z¦ d dl§m¨Z¨ d dl©mªZª g d¢Z«dS )é   )Ú
functionalÚinitÚinitializerÚquantÚutils)ÚClipGradByGlobalNormÚClipGradByNormÚClipGradByValue)ÚBeamSearchDecoderÚdynamic_decode)Úloss)ÚCELUÚELUÚGELUÚGLUÚSELUÚ
HardshrinkÚHardsigmoidÚ	HardswishÚHardtanhÚ	LeakyReLUÚ
LogSigmoidÚ
LogSoftmaxÚMaxoutÚMishÚPReLUÚReLUÚReLU6ÚRReLUÚSigmoidÚSiluÚSoftmaxÚ	Softmax2DÚSoftplusÚ
SoftshrinkÚSoftsignÚSwishÚTanhÚ
TanhshrinkÚThresholdedReLU)ÚAlphaDropoutÚBilinearÚCosineSimilarityÚDropoutÚ	Dropout2DÚ	Dropout3DÚ	EmbeddingÚFeatureAlphaDropoutÚFlattenÚFoldÚIdentityÚLinearÚPad1DÚPad2DÚPad3DÚ	UnflattenÚUnfoldÚUpsampleÚUpsamplingBilinear2DÚUpsamplingNearest2DÚ	ZeroPad1DÚ	ZeroPad2DÚ	ZeroPad3D)Ú	LayerDictÚ	LayerListÚParameterDictÚParameterListÚ
Sequential)	ÚConv1DÚConv1dÚConv1DTransposeÚConv2DÚConv2dÚConv2DTransposeÚConv3DÚConv3dÚConv3DTranspose)ÚPairwiseDistance)ÚLayer)ÚAdaptiveLogSoftmaxWithLossÚBCELossÚBCEWithLogitsLossÚCosineEmbeddingLossÚCrossEntropyLossÚCTCLossÚGaussianNLLLossÚHingeEmbeddingLossÚHSigmoidLossÚ	KLDivLossÚL1LossÚMarginRankingLossÚMSELossÚMultiLabelMarginLossÚMultiLabelSoftMarginLossÚMultiMarginLossÚNLLLossÚPoissonNLLLossÚRNNTLossÚSmoothL1LossÚSoftMarginLossÚTripletMarginLossÚTripletMarginWithDistanceLoss)Ú	BatchNormÚBatchNorm1DÚBatchNorm2DÚBatchNorm3DÚ	GroupNormÚInstanceNorm1DÚInstanceNorm2DÚInstanceNorm3DÚ	LayerNormÚLocalResponseNormÚSpectralNormÚSyncBatchNorm)ÚAdaptiveAvgPool1DÚAdaptiveAvgPool2DÚAdaptiveAvgPool3DÚAdaptiveMaxPool1DÚAdaptiveMaxPool2DÚAdaptiveMaxPool3DÚ	AvgPool1DÚ	AvgPool2DÚ	AvgPool3DÚFractionalMaxPool2DÚFractionalMaxPool3DÚLPPool1DÚLPPool2DÚ	MaxPool1DÚ	MaxPool2DÚ	MaxPool3DÚMaxUnPool1DÚMaxUnPool2DÚMaxUnPool3D)	ÚGRUÚLSTMÚRNNÚBiRNNÚGRUCellÚLSTMCellÚRNNCellBaseÚ	SimpleRNNÚSimpleRNNCell)ÚMultiHeadAttentionÚTransformerÚTransformerDecoderÚTransformerDecoderLayerÚTransformerEncoderÚTransformerEncoderLayer)ÚChannelShuffleÚPixelShuffleÚPixelUnshuffle)Ú	Parameter)Úspectral_norm)’rh   r   rl   rp   rr   ri   rj   rk   rm   rn   ro   rs   rq   r0   r5   r;   r=   r<   r6   r7   r8   r,   r-   r.   r/   r+   r*   r1   r:   r3   r   r   rŒ   r‹   r‰   rŠ   rŽ   rˆ   r‡   r   r   r   r%   r‘   r]   r   r
   r   r   rO   rS   rd   rƒ   rx   r   r#   rZ   r{   r[   r   rz   rv   ry   ra   rb   rF   rG   rE   r   rH   rw   r”   r!   r"   rC   rD   rI   rJ   r$   r   r“   rU   r   r   r   r    rK   rV   rc   r)   ru   r   rP   r’   rL   rM   r'   rN   r2   rt   r(   rY   r   r•   r|   r‚   r\   rB   r	   rR   r   r   r   r   r&   r   r—   r˜   r–   r   r   rA   r?   r„   r…   r†   r_   rX   r4   rT   r   r`   r^   rg   rf   re   rW   rQ   r9   r}   r~   r   r€   r>   r@   r™   N)¬Ú r   r   r   r   r   Zclipr   r   r	   Údecoder
   r   Úlayerr   Zlayer.activationr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   Zlayer.commonr*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   Zlayer.containerrA   rB   rC   rD   rE   Z
layer.convrF   rG   rH   rI   rJ   rK   rL   rM   rN   Zlayer.distancerO   Zlayer.layersrP   Z
layer.lossrQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   r[   r\   r]   r^   r_   r`   ra   rb   rc   rd   re   rf   rg   Z
layer.normrh   ri   rj   rk   rl   rm   rn   ro   rp   rq   rr   rs   Zlayer.poolingrt   ru   rv   rw   rx   ry   rz   r{   r|   r}   r~   r   r€   r   r‚   rƒ   r„   r…   r†   Z	layer.rnnr‡   rˆ   r‰   rŠ   r‹   rŒ   r   rŽ   r   Zlayer.transformerr   r‘   r’   r“   r”   r•   Zlayer.visionr–   r—   r˜   Z	parameterr™   Zutils.spectral_norm_hookrš   Ú__all__© rŸ   rŸ   úY/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/paddle/nn/__init__.pyÚ<module>   s&   |d,d8T, 