Задача 3. Адаптивная фильтрация

import numpy as np

def adaptive_filter(
        Vs: np.ndarray,
        Vj: np.ndarray,
        diag_A: np.ndarray
) -> np.ndarray:
    K = Vj.shape[1]
    I = np.eye(K, K)
    A = np.diag(diag_A)
    VjH = Vj.conjugate().T
    part = I + VjH @ Vj @ A
    y = Vs - Vj @ np.linalg.inv(part) @ (VjH @ Vs)
    return y