follow convention for dimension. (#333)

ffcda23f · Vincent QB · GitHub · 6db2ad1c · ffcda23f
Unverified Commit ffcda23f authored Nov 06, 2019 by Vincent QB Committed by GitHub Nov 06, 2019
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchaudio/augmentations.py torchaudio/augmentations.py +1 -1

No files found.
--- a/torchaudio/augmentations.py
+++ b/torchaudio/augmentations.py
@@ -41,7 +41,7 @@ class TimeStretch(torch.jit.ScriptModule):
                If no rate is passed, use ``self.fixed_rate``

        Returns:
-            (Tensor): Stretched complex spectrogram of dimension (*, channel, n_freq, ceil(time/rate), complex=2)
+            (Tensor): Stretched complex spectrogram of dimension (*, channel, freq, ceil(time/rate), complex=2)
        """
        assert complex_specgrams.size(-1) == 2, "complex_specgrams should be a complex tensor, shape (*, complex=2)"