Update wavernn.py (#2347)

Summary:
fix false shape

Pull Request resolved: https://github.com/pytorch/audio/pull/2347

Reviewed By: carolineechen

Differential Revision: D35921047

Pulled By: nateanl

fbshipit-source-id: 5b58820ee777920c68f13a15d80cd2bcc931af87

torchaudio/models/wavernn.py

@@ -140,7 +140,7 @@ class UpsampleNetwork(nn.Module):
     Examples
         >>> upsamplenetwork = UpsampleNetwork(upsample_scales=[4, 4, 16])
         >>> input = torch.rand(10, 128, 10)  # a random spectrogram
-        >>> output = upsamplenetwork(input)  # shape: (10, 1536, 128), (10, 1536, 128)
+        >>> output = upsamplenetwork(input)  # shape: (10, 128, 1536), (10, 128, 1536)
     """
 
     def __init__(