Skip to content

Commit ffcda23

Browse files
authored
follow convention for dimension. (#333)
1 parent 6db2ad1 commit ffcda23

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchaudio/augmentations.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ def forward(self, complex_specgrams, overriding_rate=None):
4141
If no rate is passed, use ``self.fixed_rate``
4242
4343
Returns:
44-
(Tensor): Stretched complex spectrogram of dimension (*, channel, n_freq, ceil(time/rate), complex=2)
44+
(Tensor): Stretched complex spectrogram of dimension (*, channel, freq, ceil(time/rate), complex=2)
4545
"""
4646
assert complex_specgrams.size(-1) == 2, "complex_specgrams should be a complex tensor, shape (*, complex=2)"
4747

0 commit comments

Comments
 (0)