lattice/pyerrors
1
0
Fork 0
mirror of https://github.com/fjosw/pyerrors.git synced 2025-03-15 14:50:25 +01:00
Code Issues Projects Releases Packages Wiki Activity

feat: matmul method added to correlator class.

Browse source
This commit is contained in:
Fabian Joswig 2023-07-13 11:17:27 +01:00
parent 2cd076dbd7
commit 4071c8279c
No known key found for this signature in database
1 changed files with 14 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
pyerrors/correlators.py
View file

@ -1078,6 +1078,20 @@ class Corr:
else:
raise TypeError("Corr * wrong type")
def __matmul__(self, y):
if isinstance(y, np.ndarray):
if not self.N == y.shape[0]:
raise TypeError("Shape mismatch")
newcontent = []
for t in range(self.T):
if _check_for_none(self, self.content[t]):
newcontent.append(None)
else:
newcontent.append(self.content[t] @ y)
return Corr(newcontent)
else:
raise NotImplementedError("Matmul not implemented for this type.")
def __truediv__(self, y):
if isinstance(y, Corr):
if not ((self.N == 1 or y.N == 1 or self.N == y.N) and self.T == y.T):