code: def remove_above_nyquist(amplitudes, pitch, sampling_rate):
      n_harm = amplitudes.shape[-1]
      pitches = pitch * torch.arange(1, n_harm + 1).to(pitch)
      aa = (pitches < sampling_rate / 2).float() + 1e-4
      return amplitudes * aa
  
imports: import math
  import librosa
  import numpy
  import torch
  import torch.fft
  import torch.nn
  from scipy.signal import get_window
label_desc: Use image node_id 6 for calling a global function with example usage: remove_above_nyquist(amplitudes, pitch, sampling_rate) and returns: unknown