polishing

This commit is contained in:
Martin Diehl 2020-04-20 10:13:15 +02:00
parent e06251e53e
commit bce5ed62d5
1 changed files with 4 additions and 4 deletions

View File

@ -146,7 +146,7 @@ def cell_displacement_avg(size,F):
"""
F_avg = _np.average(F,axis=(0,1,2))
return _np.einsum('ml,ijkl->ijkm',F_avg-_np.eye(3),cell_coord0(F.shape[:3][::-1],size))
return _np.einsum('ml,ijkl->ijkm',F_avg - _np.eye(3),cell_coord0(F.shape[:3][::-1],size))
def cell_displacement(size,F):
@ -282,7 +282,7 @@ def node_displacement_avg(size,F):
"""
F_avg = _np.average(F,axis=(0,1,2))
return _np.einsum('ml,ijkl->ijkm',F_avg-_np.eye(3),node_coord0(F.shape[:3][::-1],size))
return _np.einsum('ml,ijkl->ijkm',F_avg - _np.eye(3),node_coord0(F.shape[:3][::-1],size))
def node_displacement(size,F):