Adrien/QElephant
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

updated Matrix with numpy

Browse source
This commit is contained in:
Didictateur 2023-10-23 14:36:21 +02:00
parent 2b5b92c9f8
commit 4dd7c77c20
1 changed files with 2 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
QElephant/Matrix.py
View file

@ -23,17 +23,8 @@ class Matrix:
def __mul__(self, __value: "Matrix") -> "Matrix":
if type(__value) is not Matrix:
raise ValueError(f"cannot multiply Matrix with {type(__value)}")
l: list[list[complex]] = []
for i in range(self.__size[0]*__value.__size[0]):
l.append([0]*(self.__size[1]*__value.__size[1]))
for i in range(self.__size[0]):
for j in range(self.__size[1]):
for x in range(__value.__size[0]):
for y in range(__value.__size[1]):
pos = (i*__value.__size[0]+x, j*__value.__size[1]+y)
l[pos[0]][pos[1]] = self.__m[i][j]*__value.__m[x][y]
l = np.kron(self.__m, __value._Matrix__m).tolist()
return Matrix(l)
@ -46,10 +37,7 @@ class Matrix:
if not self.__size[1] == len(x):
ValueError(f"the vector was expectedd to be of size {self.__size[1]}, but is of size {len(x)}")
y: list[complex] = [0]*self.__size[0]
for i in range(self.__size[0]):
for j in range(self.__size[1]):
y[i] += self.__m[i][j]*x[j]
y: list[complex] = np.dot(self.__m, x).tolist()
return y
def __str__(self) -> str: