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

Mask estimator class.

Initialize mask estimator.

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

Calculate Mask estimator forward propagation.

• 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)