torchsig.transforms.functional.spectrogram

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

Computes spectrogram from IQ data. Directly uses compute_spectrogram inside of utils/dsp.py.

Parameters:

  • data (np.ndarray) – IQ samples.

  • fft_size (int) – The FFT size (number of bins) in the spectrogram.

  • fft_stride (int) – The number of data points to move or “hop” over when computing the next FFT.

  • rng (np.random.Generator) – Optional random generator.

Returns:

Spectrogram computed from IQ data.

Return type:

np.ndarray