Upload core.py

librosa/core.py

@@ -0,0 +1,27 @@
+import numpy as np
+import tensorflow as tf
+
+def stft(y, n_fft=2048, hop_length=512, win_length=2048, **kwargs):
+    y = tf.convert_to_tensor(y, dtype=tf.float32)
+    Z = tf.signal.stft(
+        y,
+        frame_length=win_length,
+        frame_step=hop_length,
+        fft_length=n_fft,
+        window_fn=tf.signal.hann_window
+    )
+    return Z
+
+def istft(Z, hop_length=512, win_length=2048, **kwargs):
+    if not tf.executing_eagerly():
+        tf.config.run_functions_eagerly(True)
+
+    Z = tf.convert_to_tensor(Z, dtype=tf.complex64)
+
+    audio = tf.signal.inverse_stft(
+        Z,
+        frame_length=win_length,
+        frame_step=hop_length,
+        window_fn=tf.signal.hann_window
+    )
+    return audio.numpy()