pyFDN.matrix_sqrt

pyFDN.matrix_sqrt(A)

Matrix square root via eigenvalue decomposition.

sqrtm(A) = V @ sqrt(D) @ V^(-1) where A = V @ D @ V^(-1).

Parameters:

A (torch.Tensor) – Square matrix (real, will be cast to complex for eig).

Returns:

Real matrix square root of A.

Return type:

torch.Tensor