espnet2.layers.label_aggregation.LabelAggregate

source

class espnet2.layers.label_aggregation.LabelAggregate(win_length: int = 512, hop_length: int = 128, center: bool = True)

Bases: Module

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

extra_repr()

Set the extra representation of the module

To print customized extra information, you should re-implement this method in your own modules. Both single-line and multi-line strings are acceptable.

forward(input: Tensor, ilens: Tensor | None = None) → Tuple[Tensor, Tensor | None]

LabelAggregate forward function.

• Parameters:
  • input – (Batch, Nsamples, Label_dim)
  • ilens – (Batch)
• Returns: (Batch, Frames, Label_dim)
• Return type: output