functional_mask_along_axis {torchaudio}

Mask Along Axis (functional)

Description

Apply a mask along axis. Mask will be applied from indices ⁠[v_0, v_0 + v)⁠, where v is sampled from uniform (0, mask_param), and v_0 from uniform(0, max_v - v). All examples will have the same mask interval.

Usage

functional_mask_along_axis(specgram, mask_param, mask_value, axis)

Arguments

specgram

(Tensor): Real spectrogram (channel, freq, time)

mask_param

(int): Number of columns to be masked will be uniformly sampled from ⁠[0, mask_param]⁠

mask_value

(float): Value to assign to the masked columns

axis

(int): Axis to apply masking on (2 -> frequency, 3 -> time)

Value

Tensor: Masked spectrogram of dimensions (channel, freq, time)