lattice/pyerrors
1
0
Fork 0
mirror of https://github.com/fjosw/pyerrors.git synced 2025-05-16 20:43:41 +02:00
Code Issues Projects Releases Packages Wiki Activity

sinh effective mass implemented

Browse source
This commit is contained in:
Fabian Joswig 2021-10-12 11:24:43 +01:00
parent 66c443fd63
commit 117a925775
4 changed files with 28 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

9
pyerrors/linalg.py
View file

@ -12,11 +12,14 @@ from functools import partial
from autograd.extend import defvjp
_dot = partial(anp.einsum, '...ij,...jk->...ik')
# batched diag
_diag = lambda a: anp.eye(a.shape[-1]) * a
def _diag(a):
return anp.eye(a.shape[-1]) * a
# batched diagonal, similar to matrix_diag in tensorflow
def _matrix_diag(a):
reps = anp.array(a.shape)
reps[:-1] = 1