espnet2.enh.separator.tfgridnetv2_separator.GridNetV2Block

source

class espnet2.enh.separator.tfgridnetv2_separator.GridNetV2Block(emb_dim, emb_ks, emb_hs, n_freqs, hidden_channels, n_head=4, approx_qk_dim=512, 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]