o ´‹ jD ã@s,ddlZddlmZGdd„dejjƒZdS)éN)Ú functionalcs<eZdZdZ         d‡fd d „ Zd d „Z‡ZS)Ú ResidualBlockz!Residual block module in WaveNet.éé@é€éPçéTFc sÈtƒ ¡||_| r|d|} n|dddksJdƒ‚|dd|} | |_tjj|||| ||d|_|dkrEtjj||ddd|_nd|_|d} tjj| |d|d|_ tjj| |d|d|_ dS)Nr érz$Not support even number kernel size.)ÚpaddingÚdilationÚbiasF)r ) ÚsuperÚ__init__ÚdropoutÚuse_causal_convÚtorchÚnnÚConv1dÚconvÚ conv1x1_auxÚ conv1x1_outÚ conv1x1_skip) ÚselfÚ kernel_sizeÚ res_channelsÚ gate_channelsÚ skip_channelsÚ aux_channelsrr r rr Úgate_out_channels©Ú __class__©úV/home/kuhnn/.local/lib/python3.10/site-packages/TTS/vocoder/layers/parallel_wavegan.pyrs  ÿzResidualBlock.__init__c Csò|}tj||j|jd}| |¡}|jr$|dd…dd…d| d¡…fn|}d}|j| |¡d|d\}}|dur]|jdusAJ‚| |¡}|j| |¡d|d\}}||||}}t  |¡t  |¡}|  |¡} |  |¡|d}|| fS)z; x: B x D_res x T c: B x D_aux x T )ÚpÚtrainingNéÿÿÿÿr r )ÚdimgÐ?) ÚFrr%rrÚsizeÚsplitrrÚtanhÚsigmoidrr) rÚxÚcÚresidualÚsplitdimÚxaÚxbÚcaÚcbÚsr"r"r#Úforward.s *  zResidualBlock.forward) rrrrrrr TF)Ú__name__Ú __module__Ú __qualname__Ú__doc__rr6Ú __classcell__r"r"r r#rsö&r)rÚtorch.nnrr(rÚModulerr"r"r"r#Ús