o
    wZŽh^  ã                   @   sR  d dl mZ d dlmZmZmZ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& d dl'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z. d dl/m0Z0 d dl1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7 d dl8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZD d d	lEmFZFmGZG d d
lHmIZImJZJmKZKmLZLmMZMmNZN d dlOmPZPmQZQ d dlRmSZSmTZT d dlUmVZVmWZWmXZXmYZYmZZZm[Z[ d dl\m]Z]m^Z^m_Z_m`Z`maZambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZompZpmqZqmrZr d dlsmtZtmuZumvZvmwZwmxZx d dlymzZzm{Z{m|Z|m}Z}m~Z~mZm€Z€mZm‚Z‚mƒZƒm„Z„m…Z…m†Z†m‡Z‡mˆZˆ d dl‰mŠZŠm‹Z‹ d dlŒmZmŽZŽmZmZm‘Z‘m’Z’m“Z“m”Z”m•Z•m–Z–m—Z—m˜Z˜m™Z™mšZšm›Z›mœZœmZmžZžmŸZŸm Z  d dl¡m¢Z¢m£Z£m¤Z¤m¥Z¥m¦Z¦m§Z§m¨Z¨m©Z© d dlªm«Z«m¬Z¬ d dl­m®Z®m¯Z¯m°Z°m±Z±m²Z² d dl³m´Z´mµZµm¶Z¶ g d¢Z·e·e¸e·ƒks§J ‚dS )é   )ÚModule)ÚBilinearÚIdentityÚ
LazyLinearÚLinear)ÚCELUÚELUÚGELUÚGLUÚ
HardshrinkÚHardsigmoidÚ	HardswishÚHardtanhÚ	LeakyReLUÚ
LogSigmoidÚ
LogSoftmaxÚMishÚMultiheadAttentionÚPReLUÚReLUÚReLU6ÚRReLUÚSELUÚSigmoidÚSiLUÚSoftmaxÚ	Softmax2dÚSoftminÚSoftplusÚ
SoftshrinkÚSoftsignÚTanhÚ
TanhshrinkÚ	Threshold)ÚAdaptiveLogSoftmaxWithLoss)ÚBatchNorm1dÚBatchNorm2dÚBatchNorm3dÚLazyBatchNorm1dÚLazyBatchNorm2dÚLazyBatchNorm3dÚSyncBatchNorm)ÚChannelShuffle)Ú	ContainerÚ
ModuleDictÚ
ModuleListÚParameterDictÚParameterListÚ
Sequential)ÚConv1dÚConv2dÚConv3dÚConvTranspose1dÚConvTranspose2dÚConvTranspose3dÚ
LazyConv1dÚ
LazyConv2dÚ
LazyConv3dÚLazyConvTranspose1dÚLazyConvTranspose2dÚLazyConvTranspose3d)ÚCosineSimilarityÚPairwiseDistance)ÚAlphaDropoutÚDropoutÚ	Dropout1dÚ	Dropout2dÚ	Dropout3dÚFeatureAlphaDropout)ÚFlattenÚ	Unflatten)ÚFoldÚUnfold)ÚInstanceNorm1dÚInstanceNorm2dÚInstanceNorm3dÚLazyInstanceNorm1dÚLazyInstanceNorm2dÚLazyInstanceNorm3d)ÚBCELossÚBCEWithLogitsLossÚCosineEmbeddingLossÚCrossEntropyLossÚCTCLossÚGaussianNLLLossÚHingeEmbeddingLossÚ	HuberLossÚ	KLDivLossÚL1LossÚMarginRankingLossÚMSELossÚMultiLabelMarginLossÚMultiLabelSoftMarginLossÚMultiMarginLossÚNLLLossÚ	NLLLoss2dÚPoissonNLLLossÚSmoothL1LossÚSoftMarginLossÚTripletMarginLossÚTripletMarginWithDistanceLoss)ÚCrossMapLRN2dÚ	GroupNormÚ	LayerNormÚLocalResponseNormÚRMSNorm)ÚCircularPad1dÚCircularPad2dÚCircularPad3dÚConstantPad1dÚConstantPad2dÚConstantPad3dÚReflectionPad1dÚReflectionPad2dÚReflectionPad3dÚReplicationPad1dÚReplicationPad2dÚReplicationPad3dÚ	ZeroPad1dÚ	ZeroPad2dÚ	ZeroPad3d)ÚPixelShuffleÚPixelUnshuffle)ÚAdaptiveAvgPool1dÚAdaptiveAvgPool2dÚAdaptiveAvgPool3dÚAdaptiveMaxPool1dÚAdaptiveMaxPool2dÚAdaptiveMaxPool3dÚ	AvgPool1dÚ	AvgPool2dÚ	AvgPool3dÚFractionalMaxPool2dÚFractionalMaxPool3dÚLPPool1dÚLPPool2dÚLPPool3dÚ	MaxPool1dÚ	MaxPool2dÚ	MaxPool3dÚMaxUnpool1dÚMaxUnpool2dÚMaxUnpool3d)ÚGRUÚGRUCellÚLSTMÚLSTMCellÚRNNÚRNNBaseÚRNNCellÚRNNCellBase)Ú	EmbeddingÚEmbeddingBag)ÚTransformerÚTransformerDecoderÚTransformerDecoderLayerÚTransformerEncoderÚTransformerEncoderLayer)ÚUpsampleÚUpsamplingBilinear2dÚUpsamplingNearest2d)¡r}   r~   r   r$   r€   r   r‚   rA   rƒ   r„   r…   rQ   rR   r%   r&   r'   r   r   rU   r,   rl   rm   rn   ro   rp   rq   r-   r3   r4   r5   r6   r7   r8   rS   r?   rT   rg   rB   rC   rD   rE   r   r™   rš   rF   rG   rI   r†   r‡   r	   r
   r‘   r’   rV   rh   r   r   r   r   rW   rX   r   rK   rL   rM   rY   rZ   rˆ   r‰   rŠ   r“   r”   ri   r(   r)   r*   r9   r:   r;   r<   r=   r>   rN   rO   rP   r   r   r   rj   r   r   r\   r[   r‹   rŒ   r   rŽ   r   r   r   r   r.   r/   r]   r^   r_   r   r`   ra   r   r@   r0   r1   r{   r|   rb   rk   r•   r–   r—   r˜   r   r   r   rr   rs   rt   ru   rv   rw   r   r2   r   r   rc   rd   r   r   r   r   r   r    r+   r!   r"   r#   r›   rœ   r   rž   rŸ   re   rf   rH   rJ   r    r¡   r¢   rx   ry   rz   N)¹Úmoduler   Zlinearr   r   r   r   Z
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adaptiver$   Z	batchnormr%   r&   r'   r(   r)   r*   r+   Zchannelshuffler,   Ú	containerr-   r.   r/   r0   r1   r2   Úconvr3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   Zdistancer?   r@   ZdropoutrA   rB   rC   rD   rE   rF   ÚflattenrG   rH   ÚfoldrI   rJ   ZinstancenormrK   rL   rM   rN   rO   rP   ZlossrQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   r[   r\   r]   r^   r_   r`   ra   rb   rc   rd   re   rf   Znormalizationrg   rh   ri   rj   rk   Úpaddingrl   rm   rn   ro   rp   rq   rr   rs   rt   ru   rv   rw   rx   ry   rz   Zpixelshuffler{   r|   Zpoolingr}   r~   r   r€   r   r‚   rƒ   r„   r…   r†   r‡   rˆ   r‰   rŠ   r‹   rŒ   r   rŽ   r   r   Zrnnr‘   r’   r“   r”   r•   r–   r—   r˜   Úsparser™   rš   Ztransformerr›   rœ   r   rž   rŸ   Z
upsamplingr    r¡   r¢   Ú__all__Úsorted© r¬   r¬   úH/var/www/auris/lib/python3.10/site-packages/torch/nn/modules/__init__.pyÚ<module>   s2    |$	 8  `DX( &