YouTipThe output result is:
Matrix shape: torch.Size([100, 100])Eigenvalues: tensor([105.0154, 101.8972, 98.6843])
Pytorch Torch Lobpcg
# LOBPCG Solve
eigenvalues, eigenvectors = torch.lobpcg(A, X=X, k=k)
print("Matrix shape:", A.shape)
print("Eigenvalue:")
print(eigenvalues)