cupyx.scipy.sparse.linalg.lobpcg#

cupyx.scipy.sparse.linalg.lobpcg(A, X, B=None, M=None, Y=None, tol=None, maxiter=None, largest=True, verbosityLevel=0, retLambdaHistory=False, retResidualNormsHistory=False)[source]#

Locally Optimal Block Preconditioned Conjugate Gradient Method (LOBPCG)

LOBPCG is a preconditioned eigensolver for large symmetric positive definite (SPD) generalized eigenproblems.

Parameters:
  • A (array-like) – The symmetric linear operator of the problem, usually a sparse matrix. Can be of the following types - cupy.ndarray - cupyx.scipy.sparse.csr_matrix - cupy.scipy.sparse.linalg.LinearOperator

  • X (cupy.ndarray) – Initial approximation to the k eigenvectors (non-sparse). If A has shape=(n,n) then X should have shape shape=(n,k).

  • B (array-like) – The right hand side operator in a generalized eigenproblem. By default, B = Identity. Can be of following types: - cupy.ndarray - cupyx.scipy.sparse.csr_matrix - cupy.scipy.sparse.linalg.LinearOperator

  • M (array-like) – Preconditioner to A; by default M = Identity. M should approximate the inverse of A. Can be of the following types: - cupy.ndarray - cupyx.scipy.sparse.csr_matrix - cupy.scipy.sparse.linalg.LinearOperator

  • Y (cupy.ndarray) – n-by-sizeY matrix of constraints (non-sparse), sizeY < n The iterations will be performed in the B-orthogonal complement of the column-space of Y. Y must be full rank.

  • tol (float) – Solver tolerance (stopping criterion). The default is tol=n*sqrt(eps).

  • maxiter (int) – Maximum number of iterations. The default is maxiter = 20.

  • largest (bool) – When True, solve for the largest eigenvalues, otherwise the smallest.

  • verbosityLevel (int) – Controls solver output. The default is verbosityLevel=0.

  • retLambdaHistory (bool) – Whether to return eigenvalue history. Default is False.

  • retResidualNormsHistory (bool) – Whether to return history of residual norms. Default is False.

Returns:

  • w (cupy.ndarray): Array of k eigenvalues

  • v (cupy.ndarray) An array of k eigenvectors. v has the same shape as X.

  • lambdas (list of cupy.ndarray): The eigenvalue history, if retLambdaHistory is True.

  • rnorms (list of cupy.ndarray): The history of residual norms, if retResidualNormsHistory is True.

Return type:

tuple

Note

If both retLambdaHistory and retResidualNormsHistory are True the return tuple has the following format (lambda, V, lambda history, residual norms history).