We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent adef7b9 commit 04e6847Copy full SHA for 04e6847
torchaudio/transforms.py
@@ -723,8 +723,9 @@ def forward(self, waveform: Tensor) -> Tensor:
723
Tensor: Tensor of audio of dimension (..., time).
724
"""
725
waveform_length = waveform.size()[-1]
726
-
727
- return self._fade_in(waveform_length) * self._fade_out(waveform_length) * waveform
+ device = waveform.device
+ return self._fade_in(waveform_length).to(device) * \
728
+ self._fade_out(waveform_length).to(device) * waveform
729
730
def _fade_in(self, waveform_length: int) -> Tensor:
731
fade = torch.linspace(0, 1, self.fade_in_len)
0 commit comments