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