GPU tensor support

[Isaac Kargar]

Hey. Is there any plan to add gpu support to do augmentation on GPU?

WavAugment/augment/effects.py

Line 138 in 54afcdb

 if not torch.is_tensor(input_tensor) or input_tensor.is_cuda: