espnet2.enh.separator.tfgridnetv3_separator.GridNetV3Block

source

class espnet2.enh.separator.tfgridnetv3_separator.GridNetV3Block(emb_dim, emb_ks, emb_hs, hidden_channels, n_head=4, qk_output_channel=4, activation='prelu', eps=1e-05)

Bases: Module

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

forward(x)

GridNetV2Block Forward.

• Parameters:
  • x – [B, C, T, Q]
  • out – [B, C, T, Q]