espnet2.diar.layers.tcn_nomask.Chomp1d

source

class espnet2.diar.layers.tcn_nomask.Chomp1d(chomp_size)

Bases: Module

To ensure the output length is the same as the input.

Initializes internal Module state, shared by both nn.Module and ScriptModule.

forward(x)

Forward.

• Parameters:x – [M, H, Kpad]
• Returns: [M, H, K]