torchaudio.functional.mask_along_axis

torchaudio.functional.mask_along_axis(specgram: Tensor, mask_param: int, mask_value: float, axis: int, p: float = 1.0) → Tensor

沿著 axis 套用遮罩。

遮罩將從索引 [v_0, v_0 + v) 應用，其中 v 從 uniform(0, max_v) 採樣，而 v_0 從 uniform(0, specgram.size(axis) - v) 採樣。當 p = 1.0 時，max_v = mask_param；否則 max_v = min(mask_param, floor(specgram.size(axis) * p))。所有樣本將具有相同的遮罩間隔。

參數:

• specgram (Tensor) – 實際聲譜圖 (…, freq, time)，至少具有 2 個維度。

• mask_param (int) – 要遮罩的列數將從 [0, mask_param] 中均勻採樣

• mask_value (float) – 要分配給被遮罩列的值

• axis (int) – 應用遮罩的軸，應為最後兩個維度之一。

• p (float, optional) – 可以遮罩的最大列比例。（預設值：1.0）

返回:

具有與輸入 specgram Tensor 相同維度的遮罩聲譜圖

返回類型:

Tensor