espnet2.enh.layers.tcn.DepthwiseSeparableConv
Less than 1 minute
espnet2.enh.layers.tcn.DepthwiseSeparableConv
class espnet2.enh.layers.tcn.DepthwiseSeparableConv(in_channels, out_channels, skip_channels, kernel_size, stride, padding, dilation, norm_type='gLN', causal=False)
Bases: Module
Initializes internal Module state, shared by both nn.Module and ScriptModule.
forward(x)
Forward.
- Parameters:x – [M, H, K]
- Returns: [M, B, K] skip_out: [M, Sc, K]
- Return type: res_out