b27721ab556fabb2a7e7fcb223ec1f9f6eb4418d,nt/speech_enhancement/beamformer.py,,get_pca_vector,#Any#,31

Before Change

    beamforming_vector = np.empty((bins, sensors), dtype=np.complex)
    for f in range(bins):
        eigenvals, eigenvecs = eigh(target_psd_matrix[f, :, :])
        beamforming_vector[f, :] = eigenvecs[:, np.argmax(eigenvals)]
    return beamforming_vector

After Change

    // Calculate eigenvals/vecs
    eigenvals, eigenvecs = np.linalg.eigh(target_psd_matrix)
    // Find max eigenvals
    vals = np.argmax(eigenvals, axis=-1)
    // Select eigenvec for max eigenval
    beamforming_vector = np.array([eigenvecs[i, :, vals[i]] for i in range(eigenvals.shape[0])])
    // Reconstruct original shape
    beamforming_vector = np.reshape(beamforming_vector, shape[:-1])

    return beamforming_vector

Instances