torchsig.transforms.functional.spectrogram¶

torchsig.transforms.functional.spectrogram(data: ndarray, fft_size: int, fft_stride: int) → ndarray[source]¶

Computes spectrogram from IQ data.

This function computes the spectrogram by applying the Short-Time Fourier Transform (STFT) to the input IQ data.

Directly uses compute_spectrogram inside of utils/dsp.py.

Parameters:

data – IQ samples.
fft_size – The FFT size (number of bins) in the spectrogram.
fft_stride – The number of data points to move or “hop” over when computing the next FFT.

Returns:

Spectrogram computed from IQ data.