Skip to content

FFT/STFT support #50

@happyTonakai

Description

@happyTonakai

I found that 2D FFT and IFFT are implemented but 1D FFT and IFFT are not, as well as STFT and ISTFT, which are very important tool in audio signal area. It would be great if nobuco can support them. Thank you for the great work!

Unimplemented nodes:
 I  File "/home/user/miniconda3/envs/mss/lib/python3.10/site-packages/nobuco/trace/trace.py", line 460 
 D  File "/home/user/workspace/test/test.py", line 224  
MyModel[model](float32_0<1,2,32256>) -> float32_7<1,2,2,32256>
 │   I  File "/home/user/workspace/test/model.py", line 89 
 │  angle[torch](complex64_1<1,2,1025,64>) -> float32_2<1,2,1025,64>
 │   I  File "/home/user/workspace/test/model.py", line 283 
 │   D  File "/home/user/workspace/test/model.py", line 320  
 ├· ISTFT[model](complex64_3<1,2,2,1025,64>) -> float32_7<1,2,2,32256>
 │   │   I  File "/home/user/workspace/test/model.py", line 331 
 └   │  istft[torch](complex64_4<4,1025,64>, window=float32_5<2048>, n_fft=2048, hop_length=512, win_length=2048, center=True) -> float32_6<4,32256>

torch.stft and istft are equivalent to tf.signal.stft and tf.signal.inverse_stft

Metadata

Metadata

Assignees

No one assigned

    Labels

    help wantedExtra attention is needed

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions