espnet2.layers.mask_along_axis.MaskAlongAxis

source

class espnet2.layers.mask_along_axis.MaskAlongAxis(mask_width_range: int | Sequence[int] = (0, 30), num_mask: int = 2, dim: int | str = 'time', replace_with_zero: 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(spec: Tensor, spec_lengths: Tensor | None = None)

Forward function.

• Parameters:spec – (Batch, Length, Freq)