o i @s4dZddlmZmZddlZGdddejjZdS)znResidual stack module in MelGAN. This code is modified from https://github.com/kan-bayashi/ParallelWaveGAN. )AnyDictNcs~eZdZdZdddddddid ifd ed ed ed ededeeefdedeeefffdd Z de j de j fddZ Z S) ResidualStackz+Residual stack module introduced in MelGAN. T LeakyReLUnegative_slopeg?ReflectionPad1d kernel_sizechannelsdilationbiasnonlinear_activationnonlinear_activation_paramspad pad_paramsc st|dddksJdtjttj|di|ttj||dd|fi|tjj|||||dttj|di|tjj||d|d|_tjj||d|d|_dS) aInitialize ResidualStack module. Args: kernel_size (int): Kernel size of dilation convolution layer. channels (int): Number of channels of convolution layers. dilation (int): Dilation factor. 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. pad (str): Padding function module name before dilated convolution layer. pad_params (Dict[str, Any]): Hyperparameters for padding function. rrz$Not support even number kernel size.)r r)rN) super__init__torchnn SequentialgetattrConv1dstack skip_layer) selfr r r rrrrr __class__rY/home/ubuntu/.local/lib/python3.10/site-packages/espnet2/gan_tts/melgan/residual_stack.pyrs "  zResidualStack.__init__creturncCs||||S)zCalculate forward propagation. Args: c (Tensor): Input tensor (B, channels, T). Returns: Tensor: Output tensor (B, chennels, T). )rr)rr"rrr!forward<s zResidualStack.forward)__name__ __module__ __qualname____doc__intboolstrrrrrTensorr$ __classcell__rrrr!rs8  *r)r(typingrrrrModulerrrrr!s