espnet2.gan_tts.parallel_wavegan.upsample.UpsampleNetwork

source

class espnet2.gan_tts.parallel_wavegan.upsample.UpsampleNetwork(upsample_scales: List[int], nonlinear_activation: str | None = None, nonlinear_activation_params: Dict[str, Any] = {}, interpolate_mode: str = 'nearest', freq_axis_kernel_size: int = 1)

Bases: Module

Upsampling network module.

Initialize UpsampleNetwork module.

• Parameters:
  • upsample_scales (List *[*int ]) – List of upsampling scales.
  • nonlinear_activation (Optional *[*str ]) – Activation function name.
  • nonlinear_activation_params (Dict *[*str , Any ]) – Arguments for the specified activation function.
  • interpolate_mode (str) – Interpolation mode.
  • freq_axis_kernel_size (int) – Kernel size in the direction of frequency axis.

forward(c: Tensor) → Tensor

Calculate forward propagation.

• Parameters:c – Input tensor (B, C, T_feats).
• Returns: Upsampled tensor (B, C, T_wav).
• Return type: Tensor