From 0986eebf7155d474a594485297fda118dfd32cf3 Mon Sep 17 00:00:00 2001
From: Bingcheng Hu <bingcheng1998@qq.com>
Date: Tue, 26 Apr 2022 06:01:53 -0700
Subject: [PATCH] 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 | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torchaudio/models/wavernn.py b/torchaudio/models/wavernn.py
index 447f2f3a85..ec8c0d88c3 100644
--- a/torchaudio/models/wavernn.py
+++ b/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__(