o i @s8dZddlmZmZmZddlZGdddejjZdS)zmHiFiGAN Residual block modules. This code is modified from https://github.com/kan-bayashi/ParallelWaveGAN. )AnyDictListNcsxeZdZdZddgddddddifd ed ed eed ed ededeee fffdd Z de j de j fddZ ZS) ResidualBlockz!Residual block module in HiFiGAN.i)rT LeakyReLUnegative_slopeg? kernel_sizechannels dilationsbiasuse_additional_convsnonlinear_activationnonlinear_activation_paramsc st||_tj|_|rtj|_|ddks Jd|D]P}|jtjt tj|di|tjj |||d|||dd|dg7_|rr|jtjt tj|di|tjj |||dd||dddg7_q"dS)agInitialize ResidualBlock module. Args: kernel_size (int): Kernel size of dilation convolution layer. channels (int): Number of channels for convolution layer. dilations (List[int]): List of dilation factors. use_additional_convs (bool): Whether to use additional convolution layers. bias (bool): Whether to add bias parameter in convolution layers. nonlinear_activation (str): Activation function module name. nonlinear_activation_params (Dict[str, Any]): Hyperparameters for activation function. rzKernel size must be odd number.)dilationrpaddingN) super__init__rtorchnn ModuleListconvs1convs2 SequentialgetattrConv1d) selfr r r rrrrr __class__rZ/home/ubuntu/.local/lib/python3.10/site-packages/espnet2/gan_tts/hifigan/residual_block.pyrsT    zResidualBlock.__init__xreturncCsBtt|jD]}|j||}|jr|j||}||}q|S)zCalculate forward propagation. Args: x (Tensor): Input tensor (B, channels, T). Returns: Tensor: Output tensor (B, channels, T). )rangelenrrr)r r$idxxtrrr#forwardRs   zResidualBlock.forward)__name__ __module__ __qualname____doc__intrboolstrrrrrTensorr* __classcell__rrr!r#rs2 @r) r.typingrrrrrModulerrrrr#s