espnet2.enh.layers.dc_crn.GluConv2d

source

class espnet2.enh.layers.dc_crn.GluConv2d(in_channels, out_channels, kernel_size, stride, padding=0)

Bases: Module

Conv2d with Gated Linear Units (GLU).

Input and output shapes are the same as regular Conv2d layers.

Reference: Section III-B in [1]

• Parameters:
  • in_channels (int) – number of input channels
  • out_channels (int) – number of output channels
  • kernel_size (int/tuple) – kernel size in Conv2d
  • stride (int/tuple) – stride size in Conv2d
  • padding (int/tuple) – padding size in Conv2d

forward(x)

ConvGLU forward.

• Parameters:x (torch.Tensor) – (B, C_in, H_in, W_in)
• Returns: (B, C_out, H_out, W_out)
• Return type: out (torch.Tensor)