def tf_map(real, imag): sqrt_i = np.sqrt(ssd(imag.T, imag)) sqrt_r = np.sqrt(ssd(real.T, real)) stft_k = ssd(real.T, imag) return stft_k / (sqrt_i * sqrt_r)