matrix_fft.py
25c192bf2368feature_greedy_coarseni…

matrix_fft.py
View Options

			# Copyright (C) 2018 by the RROMPy authors
			#
			# This file is part of RROMPy.
			#
			# RROMPy is free software: you can redistribute it and/or modify
			# it under the terms of the GNU Lesser General Public License as published by
			# the Free Software Foundation, either version 3 of the License, or
			# (at your option) any later version.
			#
			# RROMPy is distributed in the hope that it will be useful,
			# but WITHOUT ANY WARRANTY; without even the implied warranty of
			# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
			# GNU Lesser General Public License for more details.
			#
			# You should have received a copy of the GNU Lesser General Public License
			# along with RROMPy. If not, see <http://www.gnu.org/licenses/>.
			#

			import numpy as np
			from rrompy.hfengines.base import MatrixEngineBase as MEB

			def matrixFFT():
			N = 100
			solver = MEB(verbosity = 0)
			np.random.seed(420)
			solver.setSolver("SOLVE")
			fftB = np.fft.fft(np.eye(N)) * N**-.5
			solver.nAs = 2
			solver.As = [fftB.dot(np.multiply(np.arange(1, 1 + N), fftB.conj()).T),
			- np.eye(N)]
			solver.nbs = 1
			solver.bs = [np.random.randn(N) + 1.j * np.random.randn(N)]
			return solver