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)