Python PyTorch TimeMasking用法及代码示例

本文简要介绍python语言中 torchaudio.transforms.TimeMasking 的用法。

用法:

class torchaudio.transforms.TimeMasking(time_mask_param: int, iid_masks: bool = False)

参数：

• time_mask_param(int) -掩码的最大可能长度。从 [0, time_mask_param) 统一采样的索引。

• iid_masks(bool,可选的) -是否对批处理中的每个示例/通道应用不同的掩码。 (默认值：False)此选项仅在输入张量为 4D 时适用。

将掩蔽应用于时域中的频谱图。

建议于SpecAugment[1]。

示例

>>> spectrogram = torchaudio.transforms.Spectrogram()
>>> masking = torchaudio.transforms.TimeMasking(time_mask_param=80)
>>>
>>> original = spectrogram(waveform)
>>> masked = masking(original)