espnet.nets.pytorch_backend.frontends.mask_estimator.MaskEstimator

source

class espnet.nets.pytorch_backend.frontends.mask_estimator.MaskEstimator(type, idim, layers, units, projs, dropout, nmask=1)

Bases: Module

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

forward(xs: ComplexTensor, ilens: LongTensor) → Tuple[Tuple[Tensor, ...], LongTensor]

The forward function

• Parameters:
  • xs – (B, F, C, T)
  • ilens – (B,)
• Returns: The hidden vector (B, F, C, T) masks: A tuple of the masks. (B, F, C, T) ilens: (B,)
• Return type: hs (torch.Tensor)