jit/cuda test for complex norm. (#421)

test/test_transforms.py

@@ -326,6 +326,10 @@ class Tester(unittest.TestCase):
 
         _test_script_module(transforms.Spectrogram, tensor, sample_rate, sample_rate_2)
 
+    def test_scriptmodule_ComplexNorm(self):
+        tensor = torch.rand((1, 2, 201, 2))
+        _test_script_module(transforms.ComplexNorm, tensor)
+
     def test_resample_size(self):
         input_path = os.path.join(self.test_dirpath, 'assets', 'sinewave.wav')
         waveform, sample_rate = torchaudio.load(input_path)