torchaudio.functional.mask_along_axis_iid

torchaudio.functional.mask_along_axis_iid(specgrams: 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, specgrams.size(axis) - v) 採樣，當 p = 1.0 時，max_v = mask_param，否則 max_v = min(mask_param, floor(specgrams.size(axis) * p))。

參數:

• specgrams (Tensor) – 真實聲譜圖 (…, freq, time)，至少 3 個維度。

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

• mask_value (float) – 要分配給遮罩欄位的值

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

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

返回:

遮罩後的聲譜圖，其維度與輸入 specgrams Tensor 相同`

返回類型:

Tensor