espnet2.diar.layers.tcn_nomask.TemporalBlock

source

class espnet2.diar.layers.tcn_nomask.TemporalBlock(in_channels, out_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, B, K]
• Returns: [M, B, K]