From faa1418d8bc761a670cd0d94317d92c49e5c6fa8 Mon Sep 17 00:00:00 2001 From: Vincent Quenneville-Belair Date: Wed, 6 Nov 2019 14:57:20 -0500 Subject: [PATCH] follow convention for dimension. --- torchaudio/augmentations.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchaudio/augmentations.py b/torchaudio/augmentations.py index 52c6e2091b..efa07e9fd8 100644 --- a/torchaudio/augmentations.py +++ b/torchaudio/augmentations.py @@ -41,7 +41,7 @@ def forward(self, complex_specgrams, overriding_rate=None): 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)"