diff --git a/Installation/nnAudio/utils.py b/Installation/nnAudio/utils.py index 5f9a9dd..7e12198 100644 --- a/Installation/nnAudio/utils.py +++ b/Installation/nnAudio/utils.py @@ -34,7 +34,7 @@ def uniform_distribution(r1,r2, *size, device): def extend_fbins(X): """Extending the number of frequency bins from `n_fft//2+1` back to `n_fft` by reversing all bins except DC and Nyquist and append it on top of existing spectrogram""" - X_upper = torch.flip(X[:,1:-1],(0,1)) + X_upper = X[:,1:-1].flip(1) X_upper[:,:,:,1] = -X_upper[:,:,:,1] # For the imaganinry part, it is an odd function return torch.cat((X[:, :, :], X_upper), 1) @@ -532,4 +532,4 @@ def early_downsample(sr, hop_length, n_octaves, sr = new_sr - return sr, hop_length, downsample_factor \ No newline at end of file + return sr, hop_length, downsample_factor