espnet.nets.pytorch_backend.frontends.dnn_wpe.DNN_WPE

source

class espnet.nets.pytorch_backend.frontends.dnn_wpe.DNN_WPE(wtype: str = 'blstmp', widim: int = 257, wlayers: int = 3, wunits: int = 300, wprojs: int = 320, dropout_rate: float = 0.0, taps: int = 5, delay: int = 3, use_dnn_mask: bool = True, iterations: int = 1, normalization: bool = False)

Bases: Module

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

forward(data: ComplexTensor, ilens: LongTensor) → Tuple[ComplexTensor, LongTensor, ComplexTensor]

The forward function

Notation: : B: Batch C: Channel T: Time or Sequence length F: Freq or Some dimension of the feature vector

• Parameters:
  • data – (B, C, T, F)
  • ilens – (B,)
• Returns: (B, C, T, F) ilens: (B,)
• Return type: data