code: def _stft_magnitude_full_tf(
      waveform_input,
      window_length_samples,
      hop_length_samples,
      fft_length,
  ):
      """Calculate STFT magnitude (spectrogram) using tf.signal ops."""
      stft_magnitude = tf.abs(
          tf.signal.stft(
              waveform_input,
              frame_length=window_length_samples,
              frame_step=hop_length_samples,
              fft_length=fft_length,
          ),
          name="magnitude_spectrogram",
      )
      return stft_magnitude
  
file_path: magenta/magenta/models/onsets_frames_transcription/melspec_input.py
imports: import math
  from magenta.models.onsets_frames_transcription import mfcc_mel
  import numpy
  import tensorflow.compat.v1
label_desc: Use image node_id 1 for calling a global function with example usage: _stft_magnitude_full_tf(waveform_input, window_length_samples, hop_length_samples, fft_length) and returns: stft_magnitude