espnet2.diar.decoder.linear_decoder.LinearDecoder

source

class espnet2.diar.decoder.linear_decoder.LinearDecoder(encoder_output_size: int, num_spk: int = 2)

Bases: AbsDecoder

Linear decoder for speaker diarization

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

forward(input: Tensor, ilens: Tensor)

Forward.

• Parameters:
  • input (torch.Tensor) – hidden_space [Batch, T, F]
  • ilens (torch.Tensor) – input lengths [Batch]

property num_spk