nn.linalg

Linear Algebra Utilities

linear_operator_learning.nn.linalg.sqrtmh(A)

Compute the square root of a Symmetric or Hermitian positive definite matrix or batch of matrices.

Used code from this issue.

Parameters:

A (Tensor) – Symmetric or Hermitian positive definite matrix or batch of matrices.

Return type:

Tensor

Shape:

A: \((N, N)\)

Output: \((N, N)\)