Add autograd tests for TimeMasking/FrequencyMasking#1498

Merged

mthrok merged 5 commits intopytorch:masterfrom kiri11:time_masking_test

May 12, 2021

+25

Commits on May 10, 2021

Add autograd test for TimeMasking transform
Kirill Ignatev
authored

Commits on May 11, 2021

Add wrapper and second masking transform
Kirill Ignatev
authored
Add assert for iid codepath
Kirill Ignatev
authored

Commits on May 12, 2021

Split test and use torch.cat
Kirill Ignatev
authored
Add check that tensor is 4D for iid masking
Kirill Ignatev
authored