code: def get_psd(self, omega):
          
          w = np.asarray(omega)
          (alpha_real, beta_real, alpha_complex_real, alpha_complex_imag,
           beta_complex_real, beta_complex_imag) = self.coefficients
          p = get_psd_value(
              alpha_real, beta_real,
              alpha_complex_real, alpha_complex_imag,
              beta_complex_real, beta_complex_imag,
              w.flatten(),
          )
          return p.reshape(w.shape)
docs: Compute the PSD of the term for an array of angular frequencies
  
          Args:
              omega (array[...]): An array of frequencies where the PSD should
                  be evaluated.
  
          Returns:
              The value of the PSD for each ``omega``. This will have the same
              shape as ``omega``.