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, 可选) – 可以掩码的列的最大比例。（默认值：1.0）

返回值：

与输入谱图 Tensor` 具有相同维度的掩码谱图

返回类型：

Tensor