code: def get_stft_magnitude(
      raw_data: torch.Tensor,
      sampling_rate_in_hz: int,
      window_length_in_s: float,
      window_shift_in_s: float,
      num_fft_points: int,
      window_type: str,
  ):
      stft = _get_stft(
          raw_data,
          sampling_rate_in_hz,
          window_length_in_s,
          window_shift_in_s,
          num_fft_points,
          window_type=window_type,
      )
      stft_magnitude = torch.abs(stft)
      return torch.transpose(stft_magnitude, 0, 1)
  
imports: import functools
  import logging
  from io import BytesIO
  from typing import Any, List, Optional, Union
  import torch
  import torch.nn.functional
  import torchaudio
  from packaging import version
  from ludwig.api_annotations import DeveloperAPI
  from ludwig.constants import DEFAULT_AUDIO_TENSOR_LENGTH
  from ludwig.utils.types import TorchAudioTuple
label_desc: Use image node_id 9 for calling a global function with example usage: get_stft_magnitude(raw_data, sampling_rate_in_hz, window_length_in_s, window_shift_in_s, num_fft_points, window_type) and returns: torch