espnet2.beats.utils.make_pad_mask

source

espnet2.beats.utils.make_pad_mask(lengths: Tensor, maxlen: int | None = None) → Tensor

Linear-memory pad mask. (B, maxlen) bool, True at padded positions.

Replaces espnet2.legacy …nets_utils.make_pad_mask, whose default ONNX traceable path materialises an O(maxlen^2) helper that OOMs on long raw-waveform lengths (10 s @ 16 kHz → ~25 GB).