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