Merge pull request #362 from Martin1994/np-tf-fix

Use tf.abs in compute_spectrogram_tf
This commit is contained in:
Félix Voituret
2020-05-15 15:43:41 +02:00
committed by GitHub

View File

@@ -44,7 +44,7 @@ def compute_spectrogram_tf(
periodic=True,
dtype=waveform.dtype) ** window_exponent),
perm=[1, 2, 0])
return np.abs(stft_tensor) ** spec_exponent
return tf.abs(stft_tensor) ** spec_exponent
def time_stretch(