search

google-deepmind / dm_aux

Apache License 2.0
62 stars 6 forks source link

Allow custom window functions for STFT and ISTFT #5

Open DBraun opened 2 months ago

DBraun commented 2 months ago

For istft and stft, the window_fn argument can now be a callable or a numpy/jax.numpy array. I added tests too.

Motivation: I wanted to be able to do a square-root Hann window like AudioTools can do.