espnet2.diar.layers.tcn_nomask.GlobalLayerNorm

source

class espnet2.diar.layers.tcn_nomask.GlobalLayerNorm(channel_size)

Bases: Module

Global Layer Normalization (gLN).

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

forward(y)

Forward.

• Parameters:y – [M, N, K], M is batch size, N is channel size, K is length
• Returns: [M, N, K]
• Return type: gLN_y

reset_parameters()